在pandas中，选择其标签包含指定级别的给定字符串的所有列

Question

我想写：

mydf[mydf.columns.get_level_values('level_2').str.contains('mystring')]

但这不起作用。这里的语法是什么？

Answer 1

尝试：

idx = mydf.columns.get_level_values('level_2').to_series().str.contains('mystring')
mydf.loc[:, idx.values]

或：

idx = mydf.columns.get_level_values('level_2').map(lambda t: 'mystring' in t)
mydf.loc[:, idx]