给出此数据框:
df = pd.DataFrame({'animal': ['lions', 'tigers', 'bears', 'ohmy'],
'lorem': ['dolor', 'sit', 'amet', 'hinc'],
'ipsum': ['hinc', 'habeo', 'voluptatibus', 'pri'],
'number': [10, 20, 30, 999]})
df = df.set_index('animal')
df
lorem ipsum number
animal
lions dolor hinc 10
tigers sit habeo 20
bears amet voluptatibus 30
ohmy hinc pri 999
是否有一种简洁的方法可以根据number列的最大值将索引作为字符串返回?下面的代码完全满足了我的要求,但是我无法找出一种方法来实现它(我意识到我可以摆脱maxNum分配并进行直接评估,但是感觉不到更好)。似乎应该有一种将.max()
与df.index[]
或类似的东西结合使用的方法,但是如果有的话,我无法弄清楚:
maxNum = df['number'].max()
df.index[df['number'] == maxNum][0]
'ohmy'