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)

但这不是我想要的

我想使用单个变量(并且必须通过变量)

有人可以帮忙吗？

Answer 1

drop 需要一个列表作为输入参数，因此您需要编写

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

因为排除已经是一个列表。