SVM-python中预测概率的提取

谁能告诉我如何从predict_proba SVM的结果中转换概率数据。结果如下面的示例所示，但我想要一个只包含第二列且不含括号的列表或数组

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

[[0.15941701 0.84058299]
[ 0.12190033  0.87809967]
[ 0.06293788  0.93706212]
...
[ 0.93175738  0.06824262]]

---

谢谢。

您只需要对多维数组使用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]

---

请参阅以下答案：。正在提取第二列=a[：，1]。谢谢大家。这实际上是一项简单的任务。