python - 仅将属于最大组的行保留在 pandas 中

Question

我有一个像这样的数据框:

df = pd.DataFrame({'id':['pt1','px1','t95','sx1','dc4', 'px5'],
'feature':['f7','f7', 'f7','f8','f8', 's1'],
'score':['2','3.3','4','8','4.9', '6']})  

我想要:
1-计算“feature”列中每个组的组大小
(结果为 f7: 3；f8: 2；s1:1)
2- 找到最大组大小(即 f7: 3)
3-保留属于最大大小(f7)组的行并删除其余行

最终的 pd.DataFrame 将是:

id feature score
pt1   f7      2
px1   f7     3.3
t95   f7      4

谢谢

Answer 1

使用 GroupBy + tranform 和 count 以及额外的过滤步骤:

v = df.groupby('feature').feature.transform('count')
df[v.eq(v.max())]

    id feature score
0  pt1      f7     2
1  px1      f7   3.3
2  t95      f7     4