python - 基于级别计数过滤 Pandas MultiIndex DataFrame

Question

我有一个 MultiIndex Pandas 数据框，如下所示:

col1    col2    col3
a       e       m
b       f       n
        g       o
c       h       p
        i       q
d       j       r
        k       s
        l       t

MultiIndex 由col1 和col2 组成。

如何在 col2 中选择正好包含两个索引标签的 col1 的“行”？

即，我想要:

col1    col2    col3
b       f       n
        g       o
c       h       p
        i       q

Answer 1

使用群组 filter

df.groupby('col1').filter(lambda x: len(x) == 2)

          col3
col1 col2     
b    f       n
     g       o
c    h       p
     i       q