Метод argmin класса ndarray
Метод argmin возвращает индекс минимального элемента массива. Первым параметром можно указать ось (axis) для поиска минимума вдоль определенного направления. Вторым параметром можно указать, следует ли сохранять размерность (keepdims). Возвращает индекс или массив индексов. При наличии нескольких минимальных элементов возвращает индекс первого из них.
Синтаксис
arr.argmin([axis], [keepdims])
Пример
Найдем индекс минимального элемента в одномерном массиве:
import numpy as np
arr = np.array([3, 1, 4, 1, 5, 9, 2])
idx = arr.argmin()
print("Index of min:", idx)
print("Min value:", arr[idx])
Результат выполнения кода:
Index of min: 1
Min value: 1
Пример
Найдем индексы минимальных элементов вдоль строк (ось 1):
import numpy as np
arr = np.array([[3, 1, 4], [1, 5, 9], [2, 6, 5]])
idx = arr.argmin(axis=1)
print("Indices:", idx)
Результат выполнения кода:
Indices: [1 0 0]
Пример
Найдем индексы минимальных элементов вдоль столбцов (ось 0):
import numpy as np
arr = np.array([[3, 1, 4], [1, 5, 9], [2, 6, 5]])
idx = arr.argmin(axis=0)
print("Indices:", idx)
Результат выполнения кода:
Indices: [1 0 0]
Пример
Сохранение размерности с помощью параметра keepdims:
import numpy as np
arr = np.array([[3, 1, 4], [1, 5, 9]])
idx = arr.argmin(axis=1, keepdims=True)
print("Indices with keepdims:", idx)
Результат выполнения кода:
Indices with keepdims: [[1]
[0]]
