在numpy ndarray中找到每個子數組的最大元素

Question

我有一個ndarray，其中包含與特定類相關聯的概率。 這是多類問題，其中每個記錄都可以關聯到0-4類。

我使用了sckit-learn的分類器之一：

classifier = RandomForestClassifier(n_estimators=100)
predictions_proba = classifier.predict_proba(dataframe)

讓我們看一下predictions_proba

array([[ 0.2       ,  0.36      ,  0.32      ,  0.05      ,  0.07      ],
   [ 0.04      ,  0.54      ,  0.29      ,  0.08      ,  0.05      ],
   [ 0.05      ,  0.02      ,  0.        ,  0.93      ,  0.        ],
   ..., 
   [ 0.47777778,  0.2       ,  0.13      ,  0.19      ,  0.00222222],
   [ 0.5951746 ,  0.        ,  0.        ,  0.        ,  0.4048254 ],
   [ 0.        ,  0.        ,  0.        ,  0.13837252,  0.86162748]])

我想找到在每個子數組中找到最大概率的最簡單方法。 對於上面的示例，我想返回：

[ 0.36, 0.54, 0.93,..., 0.86162748]

0.36是第一個數組中的最大概率，0.54是第二個數組中的最大概率，依此類推。

Answer 1

arr = np.array([[1,5],[7,3]])
# array([[1, 5],
#        [7, 3]])
arr.max(axis=1)
# array([5, 7])