如果存在多個數組，則查找numpy數組中最大元素的位置

Question

想知道如何找到最大值在哪裏存在多個的最大值的索引？

目前，我正在做：

max = np.amax(V) 
position = V.tolist().index(max) 

但是說我們有V = [0,1,2,5,2,5,3] 我們會得到 最大= 5 位置= 3（但不是5？）

任何人都可以幫忙嗎？ 感謝

Answer 1

您可以使用np.nonzero發現指數這等於最大值：

>>> a 
array([2, 5, 4, 4, 4, 2, 1, 1, 2, 2, 5, 4, 1, 4, 0, 0, 5, 1, 4, 1]) 
>>> np.nonzero(a == a.max())[0] 
array([ 1, 10, 16]) 
>>> a[_] 
array([5, 5, 5])