如何获取数组中所有最大值的索引

Question

在下面的数组中，有两个最大值5和5。np.argmax命令返回第一个最大值的索引。但是我想使用np.argmax获取数组中所有最大值的索引。我该怎么做？

`a= np.array([1,2,5,5,0,3])
b= np.argmax(a)
print(b)`

Answer 1

您可以按照建议的here使用np.where()：

a = np.array([1,2,5,5,0,3])
x = np.max(a)
b = np.where(a == x)[0]
print(b)