numpy.argmin

numpy.argmin(a, axis=None, out=None) [source]

Возвращает индексы минимальных значений вдоль оси.

Параметры:	a : array_like Вводной массив. axis : int, необязательно По умолчанию, индекс в сглаженном массиве, в противном случае вдоль указанной оси. out : массив, необязательно Если задано, результат будет вставлен в этот массив. Он должен быть соответствующей формы и типа.
Возвращает:	index_array : ndarray целых чисел Массив индексов в массиве. Он имеет ту же форму, что и a.shape с удаленной размерностью вдоль axis.

См. также

ndarray.argmin, argmax

amin

Минимальное значение вдоль заданной оси.

unravel_index

Преобразовать плоский индекс в кортеж индексов.

Примечания

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

Примеры

>>> a = np.arange(6).reshape(2,3)
>>> a
array([[0, 1, 2],
       [3, 4, 5]])
>>> np.argmin(a)
0
>>> np.argmin(a, axis=0)
array([0, 0, 0])
>>> np.argmin(a, axis=1)
array([0, 0])

Индексы минимальных элементов N-мерного массива:

>>> ind = np.unravel_index(np.argmin(a, axis=None), a.shape)
>>> ind
(0, 0)
>>> a[ind]
0

>>> b = np.arange(6)
>>> b[4] = 0
>>> b
array([0, 1, 2, 3, 0, 5])
>>> np.argmin(b)  # Only the first occurrence is returned.
0