numpy.argmin
numpy.argmin(a, axis=None, out=None)
Функция argmin() возвращает индекс минимального значения вдоль указанной оси.
-
- a - массив NumPy или подобный массиву объект.
- Исходный массив.
- axis - целое число (необязательный параметр).
- Определяет ось вдоль которой выполняется поиск элементов. Если равен None (по умолчанию), то поиск выполняется по сжатому до одной оси представлению исходного массива.
- out - массив NumPy (необязательный параметр).
- Позволяет сразу напрямую поместить результат в указанный массив, при условии, что он имеет подходящую форму и тип данных.
-
- ndarray - массив NumPy
- массив индексов минимальных элементов в исходном массиве a с той же формой но удаленным размером по указанной в
axisоси.
Замечание
В случаях, когда минимальному значению соответствует несколько элементов, то возвращается индекс элемента с первым вхождением.
Данная функция обладает эквивалентным методом класса 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().
