基于最大值提取值-熊猫数据帧

Question

我正试图根据另一列中的最高值('total_cases'(提取一个值('location(，并将其分配给highest_no_new

我想从中提取值的上一个数据帧称为no_new_cases

我想从no_new_cases中挑选出">total_cases"数量最多的">location

我可以访问最多的">total_cases"，但不能访问位置：

highest_no_new = no_new_cases['total_cases'].max()

请帮我返回"位置"好吗？

提前谢谢。

Answer 1

使用idxmax:

df.loc[df['total_cases'].idxmax(), 'location']

Answer 2

您可以使用iloc方法：

df['highest_no_new'] = df.iloc[df.total_cases.argmax()]['location']

示例：

>>> d = {'location': [1, 2], 'total_cases': [3, 4]}
>>> df = pd.DataFrame(d)
>>> df
location  total_cases
0         1            3
1         2            4
>>> df.iloc[df.total_cases.argmax()]['location']
2