我有一个具有以下列名的熊猫数据框架:
Result1, Test1, Result2, Test2, Result3, Test3等…
我想删除所有名称包含单词“Test”的列。这些列的数量不是静态的,而是取决于前面的函数。
我该怎么做呢?
我有一个具有以下列名的熊猫数据框架:
Result1, Test1, Result2, Test2, Result3, Test3等…
我想删除所有名称包含单词“Test”的列。这些列的数量不是静态的,而是取决于前面的函数。
我该怎么做呢?
当前回答
import pandas as pd
import numpy as np
array=np.random.random((2,4))
df=pd.DataFrame(array, columns=('Test1', 'toto', 'test2', 'riri'))
print df
Test1 toto test2 riri
0 0.923249 0.572528 0.845464 0.144891
1 0.020438 0.332540 0.144455 0.741412
cols = [c for c in df.columns if c.lower()[:4] != 'test']
df=df[cols]
print df
toto riri
0 0.572528 0.144891
1 0.332540 0.741412
其他回答
这里有一种方法:
df = df[df.columns.drop(list(df.filter(regex='Test')))]
你可以使用'filter'过滤掉你想要的列
import pandas as pd
import numpy as np
data2 = [{'test2': 1, 'result1': 2}, {'test': 5, 'result34': 10, 'c': 20}]
df = pd.DataFrame(data2)
df
c result1 result34 test test2
0 NaN 2.0 NaN NaN 1.0
1 20.0 NaN 10.0 5.0 NaN
现在过滤器
df.filter(like='result',axis=1)
得到. .
result1 result34
0 2.0 NaN
1 NaN 10.0
你可以用df。过滤器获取匹配字符串的列列表,然后使用df.drop
resdf = df.drop(df.filter(like='Test',axis=1).columns.to_list(), axis=1)
这可以在一行中完成:
df = df.drop(df.filter(regex='Test').columns, axis=1)
问题声明“我想删除名称包含单词“Test”的所有列。”
test_columns = [col for col in df if 'Test' in col]
df.drop(columns=test_columns, inplace=True)