提取python中预测_proba的概率

Question

任何人都可以帮助我告诉我如何从preadion_proba svm上的结果中转换概率数据。结果看起来像下面的典范中，但我想要一个列表或一个数组，仅包含第二列和不含括号的列表。

[[ 0.15941701  0.84058299]
 [ 0.12190033  0.87809967]
 [ 0.06293788  0.93706212]
 ...
 [ 0.93175738  0.06824262]]
<class 'numpy.ndarray'>

谢谢。

Answer 1

您只需要将numpy索引用于多维数组。在这种情况下，您需要所有第一个维度（行），仅第二个维度（第二列）中的第二个元素。因此，您使用[:, 1]

a = np.array([
    [0.15941701, 0.84058299],
    [0.12190033, 0.87809967],
    [0.06293788, 0.93706212]])
a[:,1]

给出

[ 0.84058299  0.87809967  0.93706212]