python dataframe 取每行的最大值,Python Dataframe根据列之一中的最大值选择行

I have a dataframe in python (many rows, 2 columns). I want to modify the DF with a unique value in column 1 based on the largest value in column 2 (column 2 is sorted in ascending order if that helps). I could probably write a loop but would prefer a one or two line solution. Thanks.

Ex.

ID Value

100 11

100 14

100 16

200 10

200 20

200 30

300 45

400 0

400 25

desired result

100 16

200 30

300 45

400 25

解决方案

You want to groupby on 'a' column and then get the index of the max value using idxmax and use these indices to index the orig df:

In [12]:

df.loc[df.groupby('a')['b'].idxmax()]

Out[12]:

a b

2 100 16

5 200 30

6 300 45

8 400 25