字符串“包含” - 在 Pandas MultiIndex 上切片

Question

如何根据字符串内容对 MultiIndex 进行切片？即该特定索引是否包含某个字符串？

In [12]: df = pd.DataFrame({'a': ['a', 'ab', 'b'], 
                   'c': ['d', 'd', 'd'], 
                   'val': [1, 2 , 3]}).set_index(['a', 'c'])

In [13]: df

Out[13]:

val
a   c   
a   d   1
ab  d   2
b   d   3

In [15]: df.xs('a', level='a', drop_level=False)

Out[15]:

val
a   c   
a   d   1

In[16]: df.xs(contains('a'), level='a', drop_level=False)

预期输出：

Out[16]: 

a   c   
a   d   1
ab  d   2

显然，最后一点是不可能的。

• 如何优雅地做到这一点？
• 你能以某种方式做到不区分大小写吗？

Answer 1

boolean indexing与get_level_values和 一起使用str.contains：

print (df.index.get_level_values('a'))
Index(['a', 'ab', 'b'], dtype='object', name='a'

print (df.index.get_level_values('a').str.contains('a'))
[ True  True False]

df1 = df[df.index.get_level_values('a').str.contains('a', case=False)]
print (df1)
      val
a  c     
a  d    1
ab d    2