如何使用列表和不从列表中删除数据帧列?

df = df.drop(columns=drop_cols+['var0'])

据我所知,您有一组列,希望从几个不同的数据帧中删除,同时添加另一个唯一的列,也将删除一个数据帧。您使用的命令很接近,但没有抓住要点,因为您无法以尝试的方式创建串联列表。这就是我解决问题的方法。

给定以下形式的数据帧:

    V0  V1  V2  V3
0   1   2   3   4
1   5   6   7   8
2   9   10  11  12

定义一个函数来合并colname

def mergeNames(spc_col, multi_cols):
    rslt = [spc_col]
    rslt.extend(multi+cols)
    return rslt

然后用

drop_cols = ['V1', 'V2']
df.drop(columns=mergeNames('V0', drop_cols)

产量:

    V3
0   4
1   8
2   12