过滤数据由str.contains

Question

我尝试按可能包含以下字符串的列过滤我的大数据＆＃39; io＆＃39;和＆＃39; ir＆＃39;。

DF1

index  aio   bir   ckk
1      2     3     4
2      3     4     5

我想用包含＆＃39; io＆＃39;的列创建一个新的df。和＆＃39; ir。 新的df应该看起来：

index  aio   bir  
1      2     3    
2      3     4     

我试过

df = df[:, str.contains('io','ir')] 

但我收到错误提示类型对象＆＃39; str＆＃39;没有属性＆＃39;包含＆＃39;

Answer 1

pd.DataFrame.filter

df.filter(regex='i(o|r)')

       aio  bir
index          
1        2    3
2        3    4

如果您有匹配的事项列表

things = ['io', 'ir']
df.filter(regex='|'.join(things))

       aio  bir
index          
1        2    3
2        3    4

替代

df.filter(regex='io|ir')

df.loc[:, df.columns.str.contains('io|ir')]

Answer 2

您提到str.contains

df.loc[:,df.columns.str.contains('io|ir')]
Out[354]: 
       aio  bir
index          
1        2    3
2        3    4