創建一個帶有未定義數量參數的定義

Question

將如下代碼轉換為能夠接受我們想要的盡可能多的數據幀的最佳方法是什么？

def q_grab(df, df2, df3, q): #accepts three dataframes and a column name. Looks up column in all dataframes and combine to one
    data = df[q], df2[q], df3[q]
    headers = [q+"_1", q+"_2", q+"_3"]
    data2 = pd.concat(data, axis = 1, keys=headers)
    return data2

q = 'covid_condition'
data2 = q_grab(df, df2, df3, q) #If I run function pid_set first, it will create new df based on pID it looks like

Answer 1

一種方法是使用 * 運算符來獲取參數列表（但命名您的最終參數，因此它不是列表的一部分）：

像這樣的東西：

def q_grab(*dfs, q=None): # q is a named argument to signal end of positional arguments
    data = [df[q] for df in dfs]
    headers = [q+"_"+str(i) for i in range(len(dfs))]
    data2 = pd.concat(data, axis = 1, keys=headers)
    return data2

q = 'covid_condition'
data2 = q_grab(df, df2, df3, q=q)    

一個可能更簡潔的替代方法是繼續傳遞數據幀列表作為第一個參數：

 def q_grab(dfs,q):
   

調用：

 data2 = q.grab([df,df2,df3], q)

使用上面的函數代碼