python - 从 MultiIndex 中的索引列获取唯一值

Question

我知道我可以通过重置索引来获取 DataFrame 的唯一值，但是有没有办法避免这一步并直接获取唯一值？

鉴于我有:

        C
 A B     
 0 one  3
 1 one  2
 2 two  1

我能做到:

df = df.reset_index()
uniq_b = df.B.unique()
df = df.set_index(['A','B'])

pandas 有内置的方法吗？

Answer 1

一种方法是使用 index.levels :

In [11]: df
Out[11]: 
       C
A B     
0 one  3
1 one  2
2 two  1

In [12]: df.index.levels[1]
Out[12]: Index([one, two], dtype=object)