这篇博客介绍了如何利用Pandas库在Python中查找数据框中每个组的最大值，并获取对应的最大值行的索引。通过`groupby()`和`transform()`函数的组合，可以有效地找出每个组内'count'列的最大值，并用布尔索引筛选出这些最大值的行。当一个组中有多个最大值时，所有这些值都会被返回。此外，还展示了如何新增一列'count_max'来标记每个组的最大计数值。
In [1]: df
Out[1]:
Sp Mt Value count
0 MM1 S1 a 3
1 MM1 S1 n 2
2 MM1 S3 cb 5
3 MM2 S3 mk 8
4 MM2 S4 bg 10
5 MM2 S4 dgd 1
6 MM4 S2 rd 2
7 MM4 S2 cb 2
8 MM4 S2 uyi 7
In [2]: df.groupby(['Mt'], sort=False)['count'].max()
Out[2]:
Mt
S1 3
S3 8
S4 10
S2 7
Name: count
要获得原始DF的索引，可以这样做：
In [3]: idx = df.groupby(['Mt'])['count'].transform(max) == df['count']
In [4]: df[idx]
Out[4]:
Sp Mt Value count
0 MM1 S1 a 3
3 MM2 S3 mk 8
4 MM2 S4 bg 10
8 MM4 S2 uyi 7
注意，如果每个组有多个最大值，则将返回所有值。
更新
玛丽有机会说这是行动小组的要求：
In [5]: df['count_max'] = df.groupby(['Mt'])['count'].transform(max)
In [6]: df
Out[6]:
Sp Mt Value count count_max
0 MM1 S1 a 3 3
1 MM1 S1 n 2 3
2 MM1 S3 cb 5 8
3 MM2 S3 mk 8 8
4 MM2 S4 bg 10 10
5 MM2 S4 dgd 1 10
6 MM4 S2 rd 2 7
7 MM4 S2 cb 2 7
8 MM4 S2 uyi 7 7
880
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
