如何获取熊猫数据框列的最大值并在另一列中找到相应的值？

Question

我想获取熊猫数据框列的最大值并在另一列中找到相应的值？ 假设最大值中没有重复项，因此您始终只返回一个值。 例如，

GPA   ID
2.3  Tina1
3.4  Bob1
3.6  Lia1
2.9  Tina2
4.0  Blake1
4.5  Conor2

这里的最大 GPA 是 4.5，但我想返回最大 GPA 对应的 id，所以我会返回 Conor2。 我不确定如何执行此操作，因此非常感谢您的帮助:) 谢谢！

Answer 1

您可以使用

df.loc[df.GPA == df.GPA.max(), 'ID']

你得到

5    Conor2

如果您不想要索引而只想要值，请尝试

df.loc[df.GPA == df.GPA.max(), 'ID'].values[0]

你得到

'Conor2'

Answer 2

让我们使用idxmax ：

df.set_index('ID').idxmax()

输出：

GPA    Conor2
dtype: object

具有单个值Conor2系列。