Если вы работаете над проектами по машинному обучению, вам может понадобиться найти максимальное или минимальное значение или индексы в массиве numpy. Вот тут-то и появляется библиотека numpy.
Что такое функция np.argmin() в Python?
Функция numpy.argmin() используется в Python для получения индексов минимального элемента из массива (одномерный массив) или любой строки или столбца (многомерный массив) любого заданного массива. Функция numpy argmin() принимает arr, axis и out в качестве параметров и возвращает массив.
Синтаксис
|
1 |
numpy.argmin(arr,axis=None,out=None) |
Параметры
Функция numpy argmin() принимает три аргумента:
- arr: массив, из которого мы хотим получить индексы минимального элемента.
- axis: по умолчанию это None. Но для многомерного массива, если мы собираемся найти индекс любого максимума элемента по строкам или по столбцам, мы должны указать axis = 1 или axis = 0 соответственно.
- out: если указано, результат будет вставлен в этот массив. Он должен быть соответствующей формы и типа.
Возвращаемое значение
Функция np.argmin() возвращает массив, содержащий индексы минимальных элементов.
Нахождение индекса минимального элемента одномерного массива
См. следующий код.
|
1 2 3 4 5 6 7 8 9 10 11 12 |
#Importing numpy import numpy as np #We will create an 1D array arr = np.array([4, 24, 63, 121, 4, 64]) #Printing the array print("The array is: ", arr) #Shape of the array print("Shape of the array is : ", np.shape(arr)) #Now we will print index of min value of this array print("Index of minimum value of the given array is: ", np.argmin(arr)) |
Выход
|
1 2 3 |
The array is: [ 4 24 63 121 4 64] Shape of the array is : (6,) Index of minimum value of the given array is: 0 |
Объяснение.
В этой программе мы сначала объявили массив с некоторыми случайными числами, заданными пользователем. Затем мы напечатали форму(размер) массива. Затем мы вызвали функцию argmin(), чтобы получить индекс минимального элемента из массива. Мы видим, что минимальный элемент этого массива равен 4, который находится в позиции 0, поэтому выход равен 0.
Нахождение индексов максимальных элементов многомерного массива
См. следующий код.
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
#Importing numpy import numpy as np #We will create a 2D array #Of shape 4x3 arr = np.array([(1, 9, 4),(6, 55, 4),(1, 3, 40),(5, 6, 4)]) #Printing the array print("The array is: ") print(arr) print("Shape of the array is: ", np.shape(arr)) #Now we will find the indices of minimum value for some cases #Indices of minimum value of each row a = np.argmin(arr, axis=1) print("Indices of minimum value of each row of the array is: ", a) #Indices of minimum value of each column b = np.argmin(arr, axis=0) print("Indices of minimum value of each column of the array is: ", b) |
Выход
|
1 2 3 4 5 6 7 8 |
The array is: [[ 1 9 4] [ 6 55 4] [ 1 3 40] [ 5 6 4]] Shape of the array is: (4, 3) Indices of minimum value of each row of the array is: [0 2 0 2] Indices of minimum value of each column of the array is: [0 2 0] |
Объяснение.
В этой программе мы сначала объявили матрицу размера 4×3; вы также можете увидеть форму матрицы, которая равна(4,3). Затем мы вызвали argmin(), чтобы получить вывод о различных случаях.
