熊猫:使用MultiIndex进行选择

时间:2012-11-13 02:41:12

标签: python pandas

考虑以下DataFrames

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的{​​{1}}行?

如果我在两个帧上vals,我现在可以合并/加入它们并切片但是我想要,但是如何使用(多个)索引呢?

1 个答案:

答案 0 :(得分:7)

布尔索引一直......

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