在下面的数组中，有两个最大值5和5。np.argmax命令返回第一个最大值的索引。但是我想使用np.argmax获取数组中所有最大值的索引。我该怎么做？
`a= np.array([1,2,5,5,0,3])
b= np.argmax(a)
print(b)`
您可以np.where()按照此处的建议使用：
a = np.array([1,2,5,5,0,3])
x = np.max(a)
b = np.where(a == x)[0]
print(b)
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句