使用 pandas 和 numpy 选择最大置信度类别 [重复]

【问题标题】：Selecting max confidence category using pandas and numpy [duplicate]使用 pandas 和 numpy 选择最大置信度类别 [重复]
【发布时间】：2018-09-18 20:26:47
【问题描述】：

我被困在根据数据集计算 3 列和相应类别中的最高值：

Dataset Image>

我想计算置信度的最大置信度类别并创建一个包含 4 列的数据集，例如： TC_Name 失败 MaxErrCategory MaxConfidence

我已尝试捕获每一行的最大置信度，但无法确定类别：

max_conf=data.max(axis=1)

请帮忙..

【问题讨论】：

标签： python python-3.x pandas numpy dataset

【解决方案1】：

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

data.idxmax(axis=1)

由于某些列中有字符串，因此应首先选择要计算最大值的列：

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

【讨论】：

但是我也有行中的字符串，这给我带来了错误：ValueError: could not convert string to float: 'Script'.
我编辑了我的答案以考虑您的评论。