我有一个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
组成。
如何选择col1
中仅包含col2
中两个索引标签的“行”?
col1 col2 col3
b f n
g o
c h p
i q
答案 0 :(得分:6)
df.groupby('col1').filter(lambda x: len(x) == 2)
col3
col1 col2
b f n
g o
c h p
i q
答案 1 :(得分:3)
IIUC
s=df.groupby(level=0).size()
df.loc[s[s==2].index.tolist()]
Out[583]:
col3
col1 col2
b f n
g o
c h p
i q