python取最大值所在行_pandas获取groupby分组里最大值所在的行方法

pandas获取groupby分组里最大值所在的行方法

如下面这个DataFrame,按照Mt分组，取出Count最大的那行

import pandas as pd

df = pd.DataFrame({'Sp':['a','b','c','d','e','f'], 'Mt':['s1', 's1', 's2','s2','s2','s3'], 'Value':[1,2,3,4,5,6], 'Count':[3,2,5,10,10,6]})

df

Count

Mt

Sp

Value

0

3

s1

a

1

1

2

s1

b

2

2

5

s2

c

3

3

10

s2

d

4

4

10

s2

e

5

5

6

s3

f

6

方法1：在分组中过滤出Count最大的行

df.groupby('Mt').apply(lambda t: t[t.Count==t.Count.max()])

Count

Mt

Sp

Va