Skip to main content
 首页 » 编程设计

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

2025年05月04日12shangdawei

我有一个 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 

请您参考如下方法:

使用群组 filter

df.groupby('col1').filter(lambda x: len(x) == 2) 
 
          col3 
col1 col2      
b    f       n 
     g       o 
c    h       p 
     i       q