python - Pandas：使用 MultiIndex 进行选择

Question

考虑以下数据帧

In [136]:
df = pd.DataFrame({'A':[1,1,2,2],'B':[1,2,1,2],'C':np.arange(10,30,5)}).set_index(['A','B'])
df
Out[136]:
      C
A B    
1 1  10
  2  15
2 1  20
  2  25

In [130]:
vals = pd.DataFrame({'A':[1,2],'values':[True,False]}).set_index('A')
vals
Out[130]:
  values
A       
1   True
2  False

如何仅选择 中df具有相应True值的行vals？

如果我reset_index在两个框架上，我现在可以合并/加入它们并按照我的意愿进行切片，但是我如何使用（多）索引来做到这一点？

Answer 1

布尔索引一路...

In [65]: df[pd.Series(df.index.get_level_values('A')).isin(vals[vals['values']].index)]
Out[65]: 
      C
A B    
1 1  10
  2  15

请注意，您可以在多索引上使用 xs。

In [66]: df.xs(1)
Out[66]: 
    C
B    
1  10
2  15