我有一个具有以下列名的熊猫数据框架:

Result1, Test1, Result2, Test2, Result3, Test3等…

我想删除所有名称包含单词“Test”的列。这些列的数量不是静态的,而是取决于前面的函数。

我该怎么做呢?


当前回答

这里有一种方法:

df = df[df.columns.drop(list(df.filter(regex='Test')))]

其他回答

这里有一种方法:

df = df[df.columns.drop(list(df.filter(regex='Test')))]

问题声明“我想删除名称包含单词“Test”的所有列。”

test_columns = [col for col in df if 'Test' in col]
df.drop(columns=test_columns, inplace=True)

使用数据帧。选择方法:

In [38]: df = DataFrame({'Test1': randn(10), 'Test2': randn(10), 'awesome': randn(10)})

In [39]: df.select(lambda x: not re.search('Test\d+', x), axis=1)
Out[39]:
   awesome
0    1.215
1    1.247
2    0.142
3    0.169
4    0.137
5   -0.971
6    0.736
7    0.214
8    0.111
9   -0.214

这可以在一行中完成:

df = df.drop(df.filter(regex='Test').columns, axis=1)

该方法在适当的位置执行所有操作。许多其他答案会复制,效率不高:

df.drop(df.columns[df.columns.str.contains('Test')], axis=1, inplace=True)