python - 如何在 Python 中使用单个变量删除 2 列

Question

我有这个完美运行的代码

Exclude = "A"
df = df.drop([Exclude], 1)

我想同时删除 2 列

但是这段代码不起作用

Exclude = ["A", "B"]
df = df.drop([Exclude], 1)

我知道我可以使用

Exclude1 = "A"
Exclude2 = "B"
df = df.drop([Exclude1], 1)
df = df.drop([Exclude2], 1)

但这不是我想要的

我想使用单个变量（并且必须通过变量）

谁能帮忙？

标签： python

Answer 1

drop需要一个列表作为输入参数，所以你需要写

Exclude = ["A", "B"]
df = df.drop(Exclude, 1)

因为Exclude已经是一个列表。