numpy.argmin

numpy.argmin(a, axis=None, out=None)

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

Параметры:
a - массив NumPy или подобный массиву объект.
Исходный массив.
axis - целое число (необязательный параметр).
Определяет ось вдоль которой выполняется поиск элементов. Если равен None (по умолчанию), то поиск выполняется по сжатому до одной оси представлению исходного массива.
out - массив NumPy (необязательный параметр).
Позволяет сразу напрямую поместить результат в указанный массив, при условии, что он имеет подходящую форму и тип данных.
Возвращает:
ndarray - массив NumPy
массив индексов минимальных элементов в исходном массиве a с той же формой но удаленным размером по указанной в axis оси.
Смотрите так же:
nanargmin, argmax, nanargmax

Замечание

В случаях, когда минимальному значению соответствует несколько элементов, то возвращается индекс элемента с первым вхождением.

Данная функция обладает эквивалентным методом класса ndarray, т.е. np.argmin(a) равносильно вызову метода a.argmin():

>>> a = np.random.randint(0, 20, 10)
>>> a
array([14, 10,  5, 12, 10,  9,  7,  6, 18,  7])
>>> 
>>> np.argmin(a)
2
>>> 
>>> a.argmin()
2


Примеры

>>> import numpy as np
>>> 
>>> a = np.random.randint(0, 20, size = (3, 5))
>>> a
array([[ 5, 18,  6,  3,  4],
       [ 8, 12,  9,  0,  1],
       [ 3, 12,  8, 10, 14]])
>>> 
>>> np.argmin(a)
8
>>> 
>>> a.flat[8]
0

За более подробными примерами обратитесь к странице np.argmax().