Python 使用pandas和numpy选择最大置信度类别

Python 使用pandas和numpy选择最大置信度类别,python,python-3.x,pandas,numpy,dataset,Python,Python 3.x,Pandas,Numpy,Dataset,我一直在根据数据集计算3列中的最高列和相应类别: 我想用置信水平计算最大置信度类别,并创建一个包含4列的数据集,如: TC_名称故障MaxErrCategory MaxConference 我已尝试获取每行的最大置信水平,但无法确定类别: max_conf=data.max(axis=1) 请提供帮助。数据帧的idxmax方法将为每一行提供第一次出现最大值的列的名称 data.idxmax(axis=1) 由于某些列中有字符串,应首先选择要计算最大值的列: data[ ["confidenc

我一直在根据数据集计算3列中的最高列和相应类别:

我想用置信水平计算最大置信度类别,并创建一个包含4列的数据集,如: TC_名称故障MaxErrCategory MaxConference

我已尝试获取每行的最大置信水平,但无法确定类别:

max_conf=data.max(axis=1)

请提供帮助。

数据帧的idxmax方法将为每一行提供第一次出现最大值的列的名称

data.idxmax(axis=1)
由于某些列中有字符串,应首先选择要计算最大值的列:

data[ ["confidence1", "confidence2", "confidence3"] ].idxmax(axis=1)

但我的行中也有字符串,这给我带来了错误:ValueError:无法将字符串转换为float:“Script”。我编辑了我的答案以考虑您的评论。