Как получить индекс всех максимальных значений в массиве

Question

В следующем массиве есть два максимальных значения 5 и 5. Команда np.argmax возвращает индекс первого максимального значения. Но я хочу получить индекс всех максимальных значений в массиве, используя np.argmax. Как я могу это сделать?

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

Answer 1

Вы можете использовать np.where() как предложено здесь :

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