python - 基于多索引比较选择行

Question

df = pd.DataFrame([10,20,30,40], index=[[1,1,2,2],[1,4,3,2]])
df.index.names=['a','b']

我想保留索引 a 等于索引 b 的行:

      0
1 1  10
2 2  40

我想也许 lambda 可以在这种情况下工作

df.index[lambda x: x[0]==x[1]]

但事实并非如此。我怎样才能得到上面的结果？

Answer 1

get_level_values 将为此工作

df[df.index.get_level_values(0)==df.index.get_level_values(1)]
Out[455]: 
      0
a b    
1 1  10
2 2  40

修复代码

df[df.index.map(lambda x: x[0]==x[1])]
Out[457]: 
      0
a b    
1 1  10
2 2  40