python - 字符串 "contains"- Pandas MultiIndex 上的切片

Question

如何按 MultiIndex 的字符串内容对其进行切片？ IE。该特定索引是否包含特定字符串？

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