python - python pandas通过列名称列表从数据框中选择列

Question

我有一个包含很多列的数据框。现在，我只想选择某些列。我将要选择的所有列名称保存到Python列表中，现在我想根据该列表过滤数据框。

我一直在尝试做:

df_new = df[[list]]

其中list包含我要选择的所有列名。

但是我得到了错误:

TypeError: unhashable type: 'list'

任何帮助吗？

Answer 1

您可以删除一个[]:

df_new = df[list]

更好的方法是使用其他名称作为 list，例如 L:

df_new = df[L]

看起来工作正常，我只尝试简化一下:

L = []
for x in df.columns: 
    if not "_" in x[-3:]: 
        L.append(x) 
print (L)

List comprehension:

print ([x for x in df.columns if not "_" in x[-3:]])