Python Dataframe 根据其中一列中的最大值选择行

Question

我在 python 中有一个数据框(很多行，2 列)。我想根据第 2 列中的最大值修改第 1 列中具有唯一值的 DF(如果有帮助，第 2 列按升序排序)。我可能会写一个循环，但更喜欢一两行解决方案。谢谢。

例如

ID         Value
100       11
100       14
100       16
200       10
200       20
200       30
300       45
400        0
400       25

想要的结果

100       16
200       30
300       45
400       25

Answer 1

您想groupby在“a”列上，然后使用 idxmax 获取最大值的索引并使用这些索引来索引 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