Функция argmax
Функция argmax возвращает индекс максимального значения
в массиве. Если ось не указана, поиск ведется по всему массиву.
При наличии нескольких максимальных элементов возвращается индекс
первого из них. Функция может работать как с одномерными, так
и с многомерными массивами.
Синтаксис
np.argmax(arr, [axis], [out], [keepdims])
Пример
Давайте найдем индекс максимального значения в одномерном массиве:
import numpy as np
arr = np.array([1, 5, 3, 9, 2])
res = np.argmax(arr)
print(res)
Результат выполнения кода:
3
Пример
Найдем индекс максимального элемента в двумерном массиве без указания оси:
import numpy as np
arr = np.array([[7, 2, 9], [4, 8, 6]])
res = np.argmax(arr)
print(res)
Результат выполнения кода:
2
Пример
Найдем индекс максимального элемента вдоль строк (по оси 0):
import numpy as np
arr = np.array([[7, 2, 9], [4, 8, 6]])
res = np.argmax(arr, axis=0)
print(res)
Результат выполнения кода:
[0 1 0]
Пример
Найдем индекс максимального элемента вдоль столбцов (по оси 1):
import numpy as np
arr = np.array([[7, 2, 9], [4, 8, 6]])
res = np.argmax(arr, axis=1)
print(res)
Результат выполнения кода:
[2 1]
Пример
Используем параметр keepdims для сохранения размерности результата:
import numpy as np
arr = np.array([[7, 2, 9], [4, 8, 6]])
res = np.argmax(arr, axis=1, keepdims=True)
print(res)
Результат выполнения кода:
[[2]
[1]]
Пример
Найдем индекс первого максимального элемента при наличии дублирующихся значений:
import numpy as np
arr = np.array([3, 7, 2, 7, 4])
res = np.argmax(arr)
print(res)
Результат выполнения кода:
1
