python - 从列表中指定函数中的参数

Question

我已经阅读了许多线程（这里和这里）和文档（这里和这里）。但是，我无法让它工作。我得到一个错误

AxisError: axis 0 is out of bounds for array of dimension 0

谢谢。

import pandas as pd
from scipy.stats import levene

data = {'A': [1,2,3,4,5,6,7,8],
        'B': [9,10,11,12,13,14,15,16],
        'C': [1,2,3,4,5,6,7,8]}
df3 = pd.DataFrame(data, columns=['A', 'B','C'])

print(levene(df3['A'], df3['C'])) # this works as intended

cols_of_interest = ['A','C'] # my requirement could make this any combination

# function to pass through arguments into Levene test
def func(df,cols_of_interest):
    cols = [col for col in 'df.'+df[cols_of_interest].columns] # my strategy to mimic the arguments
    lev = levene(*cols)
    print(lev)

func(df3,cols_of_interest)

标签： pythonpandasargument-unpacking

Answer 1

将您的列表理解替换def为：

cols = [df[x] for x in cols_of_interest]