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

我有一个完美的代码

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

我想同时删除两列

但是这个代码不起作用

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

我知道我可以用

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

但这不是我想要的

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

有人能帮忙吗？

改变：

至：

---

drop

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

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

---

因为

Exclude

已经是一个列表。

当您定义

Exclude=[“a”，“B”]

并调用

df.drop（[Exclude]，1）

时，您会看到您实际上正在调用

df.drop（[[“a”，“B”]]，1）

，这是一个列表。这不是drop的工作原理。因此，只需传递

排除

>>> df = pd.DataFrame({'A':list('abcde'),'B':list('ghijk'),'C':list('klmno')})
>>> Exclude = ["A", "B"]
>>> df.drop(Exclude, 1)
>>> 
   C
0  k
1  l
2  m
3  n
4  o

既然

然后

标签：单个标签或类似列表

df=df.drop（列=[Exclude]，1）^SyntaxError:位置参数跟在关键字参数后面

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

>>> df = pd.DataFrame({'A':list('abcde'),'B':list('ghijk'),'C':list('klmno')})
>>> Exclude = ["A", "B"]
>>> df.drop(Exclude, 1)
>>> 
   C
0  k
1  l
2  m
3  n
4  o

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

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