在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。
答
arr = np.array([[1,5],[7,3]])
# array([[1, 5],
# [7, 3]])
arr.max(axis=1)
# array([5, 7])
您是否在Google搜索NumPy数组中找到最大值？ – Divakar
是的。我不明白我必须将ndarray作为np.array的参数。 – Aviade
谢谢你真的帮了我。 – Aviade