如何在一行中创建一个列表并从列表中删除一个项目？

Question

假设我们有

df = pd.Dataframe(columns = ['item', 'item1', 'item2'])

我想创建一个没有一个元素的列列表。我想做类似的事情

df.columns.remove('item2') 但这样的代码只是返回“无”。同时，

l = list(df.columns)
l.remove('item2') 

工作正常。有没有办法将这段代码提炼成一行？

Answer 1

您可以使用.difference：

print(df.columns.difference(["item2"]))

打印：

Index(['item', 'item1'], dtype='object')

---

或者一个列表输出：

print(df.columns.difference(["item2"]).tolist())

打印：

['item', 'item1']