Python返回列或行最大值对应的内容

现实场景

有一堆数据,表示的是语音识别的效果(用相似度得分表示),需要找到每个测试人员最高分数对应的那个对象(判别结果)。

数据

data = {"tet":["Anna", "Anna", "Bob", "Bob"], "name":["Candy", "Duck", "Candy", "Duck" ], "scores":["254", "390", "450", "178"]}
df = pd.DataFrame(data) 
df

	tet 	name 	scores
0 	Anna 	Candy 	254
1 	Anna 	Duck 	390
2 	Bob 	Candy 	450
3 	Bob 	Duck 	178

虽然此刻能够很清楚地看到Anna与Duck的分数最高,但是如果比对的人不止这么一两个,就很难再通过肉眼观察到结果了,只能想办法找到tetscores取值最高所对应的那个name

方法

试过直接使用透视图
(Python 透视图pivot_table的用法:点击

import pandas as pd

table = pd.pivot_table(df, index='tet', aggfunc=max)
table
 	scores
tet 	
Anna 	390
Bob 	450

不过暂时不知道如何返回最大scores对应的那个name

最后成功的做法是使用groupby +dataframe 的idxmax函数:

ind = df.groupby('tet')['scores'].idxmax()
row = df.iloc[ind,:]
print(ind)
print(row)

后续

什么时候有空,对各种返回最大最小值的索引做一个总结:
python内置:max、min;
nump:argmax、argmin;
pandas:idxmax、idxmin;

参考:

  1. python中pandas的DataFrame某列最大值所在行
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值