在numpy ndarray中爲每個子陣列查找最大元素

我有一個包含與特定類關聯的概率的ndarray。這是多類問題中，每一個記錄可以關聯到類0 - 4在numpy ndarray中爲每個子陣列查找最大元素

我用的分類之一sckit學習：

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是第二陣列，等在最大probility。

來源

2016-11-17 Aviade

您是否在Google搜索NumPy數組中找到最大值？ – Divakar

是的。我不明白我必須將ndarray作爲np.array的參數。 – Aviade

謝謝你真的幫了我。 – Aviade

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

來源

2016-11-17 20:07:21

在numpy ndarray中爲每個子陣列查找最大元素

回答

相關問題