通过包含多个字符串的标头过滤Pandas Dataframe列

Question

我有一个数据框，只想显示标题包含特定字符串的列。

df.loc[:,df.columns.str.contains(['BB','TP'])]

因此，这只会显示包含BB或TP的列。但是，这不适用于2种情况。

Answer 1

使用：

L = ['BB','TP']
df.loc[:, df.columns.str.contains('|'.join(L)]