根据最大列值选择熊猫数据框索引

时间:2019-10-09 14:08:40

标签: python pandas

给出此数据框:

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'

0 个答案:

没有答案