python - 按 str.contains 过滤数据

Question

我正在尝试按可能包含以下字符串“io”和“ir”的列来过滤我的大数据。

df1

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

我想用包含“io”和“ir”的列创建一个新的 df。新的 df 应该看起来:

index  aio   bir  
1      2     3    
2      3     4     

我试过了

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

但我收到一条错误消息，提示类型对象“str”没有属性“contains”

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')]