`numpy.argmin()`

Функция numpy.argmin() возвращает индекс первого минимального элемента массива или вдоль указанной оси.

Сигнатура

numpy.argmin(a, axis=None, out=None, *, keepdims=<no value>)

Параметры

a — входной массив.
axis — ось редукции.
out — массив для записи результата.
keepdims — сохранять ли редуцированные оси.

Возвращаемое значение

Целое число (axis=None) или ndarray индексов типа intp.

Примеры

Пример 1. Индекс минимума.

import numpy as np

arr = np.array([5, 1, 9, 1, 3])
print(np.argmin(arr))

Пример 2. По осям.

m = np.array([[3, 7, 1], [8, 2, 6]])
print(np.argmin(m, axis=0))
print(np.argmin(m, axis=1))

[0 1 0]
[2 1]

Пример 3. Самый холодный час за сутки.

temps = np.array([18, 17, 16, 15, 14, 13, 12, 11, 10, 12, 15, 18,
                  21, 23, 25, 24, 22, 20, 19, 18, 17, 16, 15, 14])
hour = np.argmin(temps)
print(f"Минимум в {hour}:00 — {temps[hour]} °C")

Минимум в 8:00 — 10 °C

Пример 4. Координаты глобального минимума 2D-массива.

m = np.array([[5, 2, 9], [1, 8, 7]])
pos = np.unravel_index(np.argmin(m), m.shape)
print(pos)

(1, 0)

Пример 5. С пустым массивом нужно осторожно — выбросит исключение.

try:
    np.argmin(np.array([]))
except ValueError as e:
    print("Ошибка:", e)

Ошибка: attempt to get argmin of an empty sequence

См. также

Примечание

Лицензия и источники

Техническое описание функции адаптировано из официальной документации NumPy (https://numpy.org/doc/stable/), BSD-3-Clause License. Примеры и пояснения — © AlashEd Wiki.

numpy.argmin()