python-3.x - 如何在一行中创建一个列表并从列表中删除一个项目？

Question

假设我们有

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

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

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

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

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

标签： python-3.xlist

Answer 1

您可以使用.difference：

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

印刷：

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

---

或列表输出：

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

印刷：

['item', 'item1']