python - 熊猫，如何将列表传递到列表位置

Question

def data_cleaning(df, dict):
    df = df[df.columns[list(dict.values())]
    return df
    
    
dict = {
  "cost": 0,
  "Price": 3
}

我试图通过dictintodata_clearning() 事情是，一旦它接受列表，它就会变成df.columns[[0, 3]]一个额外的括号

如何让函数按预期运行？

最好的，

标签： pythonpandas

Answer 1

查看.iloc

def data_cleaning(df, dict):
    df = df.iloc[:,list(dict.values())]
    return df