如何使用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

既然，

Exclude="A"
df.drop(Exclude, axis=1) #works

然后

Exclude=["C", "D"]
dr.drop(Exclude, axis=1) #will work also.

每文档

标签:单标签或类似列表的标签

Answer 2

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

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

因为Exclude已经是一个列表了。

Answer 3

变更：

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

To:

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