Функция argmin
Функция argmin возвращает индекс минимального элемента
в массиве NumPy. Первым параметром функция принимает массив,
втором параметром можно указать ось, вдоль которой выполняется
поиск. Также можно передать параметр out для указания
массива, в который будет записан результат.
Синтаксис
np.argmin(arr, [axis], [out])
Пример
Давайте найдём индекс минимального элемента в одномерном массиве:
import numpy as np
arr = np.array([10, 5, 8, 3, 12])
res = np.argmin(arr)
print(res)
Результат выполнения кода:
3
Функция вернула индекс 3, так как минимальный элемент
равен 3 и находится на позиции с индексом 3.
Пример
Давайте найдём индекс минимального элемента в двумерном массиве без указания оси:
import numpy as np
arr = np.array([[4, 2, 7], [1, 9, 5]])
res = np.argmin(arr)
print(res)
Результат выполнения кода:
3
Функция вернула индекс 3, так как массив был свёрнут
в одномерный вид: [4, 2, 7, 1, 9, 5], а минимальный элемент
равен 1 и находится на позиции с индексом 3.
Пример
Давайте найдём индексы минимальных элементов вдоль оси (по строкам):
import numpy as np
arr = np.array([[4, 2, 7], [1, 9, 5]])
res = np.argmin(arr, axis=1)
print(res)
Результат выполнения кода:
[1 0]
В первой строке минимальный элемент равен 2
с индексом 1, во второй строке минимальный
элемент равен 1 с индексом 0.
Пример
Давайте найдём индексы минимальных элементов вдоль оси (по столбцам):
import numpy as np
arr = np.array([[4, 2, 7], [1, 9, 5]])
res = np.argmin(arr, axis=0)
print(res)
Результат выполнения кода:
[1 0 1]
В первом столбце минимальный элемент равен 1
с индексом 1, во втором столбце минимальный
элемент равен 2 с индексом 0, в третьем
столбце минимальный элемент равен 5 с индексом
1.
Пример
Давайте используем функцию argmin с методом
argmin объекта ndarray:
import numpy as np
arr = np.array([8, 4, 9, 1, 6])
res = arr.argmin()
print(res)
Результат выполнения кода:
3
Метод argmin вернул индекс минимального элемента
1, который находится на позиции с индексом 3.
