NumPy argmax: нахождение индекса максимального элемента в массиве

Функция numpy.argmax() возвращает индекс максимального элемента в массиве. Если массив многомерный, то индекс максимального элемента будет возвращен в виде кортежа.

Пример использования numpy.argmax() для одномерного массива:

python
import numpy as np
arr = np.array([1, 2, 3, 2, 1])
max_index = np.argmax(arr)
print("Максимальный элемент находится на позиции", max_index)  # Вывод: Максимальный элемент находится на позиции 2
print("Значение максимального элемента:", arr[max_index])  # Вывод: Значение максимального элемента: 3

В данном примере массив `arr` содержит элементы [1, 2, 3, 2, 1]. Функция argmax() возвращает индекс максимального элемента, который равен 2. Мы выводим позицию максимального элемента и его значение.

Пример использования numpy.argmax() для двумерного массива:

python
import numpy as np
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])
max_index = np.argmax(arr)
max_index_tuple = np.unravel_index(max_index, arr.shape)
print("Максимальный элемент находится на позиции", max_index_tuple)  # Вывод: Максимальный элемент находится на позиции (2, 2)
print("Значение максимального элемента:", arr[max_index_tuple])  # Вывод: Значение максимального элемента: 9

В данном примере двумерный массив `arr` содержит числа от 1 до 9. Функция argmax() возвращает позицию максимального элемента в виде одномерного индекса (значение 8). С помощью функции unravel_index() мы преобразуем одномерный индекс в позицию в двумерном массиве. Результат - кортеж (2, 2), который содержит позицию максимального элемента. Мы выводим эту позицию и значение максимального элемента.

Таким образом, функция numpy.argmax() позволяет находить индекс (или индексы в случае многомерного массива) максимального элемента в массиве. Она удобна для решения задач, связанных с поиском наибольшего элемента в массиве или определения позиции самого большого элемента.