サブ配列ごとにnumpy ndarrayの最大要素を見つける

Question

特定のクラスに関連する可能性があるndarrayがあります。 これは、すべてのレコードがクラス0に関連付けることが可能なマルチクラスの問題である - 私はsckit-学ぶの分類のいずれかを使用4.

：

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]]) 

に見てみましょう

各サブアレイで最大の確率を見つける最も簡単な方法を知りたいと思います。 I上記例えば が返すたい：

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

0.36が最初の配列で最大確率である、 0.54は二番目の配列、 などで最大probilityあります。

Answer 1

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