python - 根据列表删除熊猫列

Question

我有一个清单：

my_list = ['a', 'b']

和熊猫数据框：

d = {'a': [1, 2], 'b': [3, 4], 'c': [1, 2], 'd': [3, 4]}
df = pd.DataFrame(data=d)

我该怎么做才能df根据 listmy_list删除列，在这种情况下删除列a和b

标签： pythonpandas

Answer 1

这很简单：

df = df.drop(columns=my_list)

drop通过指定列名列表删除列