numpy.argmax() a Python
El numpy.argmax() La funció retorna índexs de l'element màxim de la matriu en un eix concret.
Sintaxi:
numpy.argmax(array, axis = None, out = None)
Paràmetres:
array : Input array to work on axis : [int, optional]Along a specified axis like 0 or 1 out : [array optional]Provides a feature to insert output to the out array and it should be of appropriate shape and dtype
Tornada:
Array of indices into the array with same shape as array.shape with the dimension along axis removed.
Codi 1:
Python
# Python Program illustrating> # working of argmax()> > import> numpy as geek> > # Working on 2D array> array> => geek.arange(> 12> ).reshape(> 3> ,> 4> )> print> (> 'INPUT ARRAY :
'> , array)> > # No axis mentioned, so works on entire array> print> (> '
Max element : '> , geek.argmax(array))> > # returning Indices of the max element> # as per the indices> print> (> '
Indices of Max element : '> , geek.argmax(array, axis> => 0> ))> print> (> '
Indices of Max element : '> , geek.argmax(array, axis> => 1> ))> |
Sortida:
INPUT ARRAY : [[ 0 1 2 3] [ 4 5 6 7] [ 8 9 10 11]] Max element : 11 Indices of Max element : [2 2 2 2] Indices of Max element : [3 3 3]
Codi 2:
Python
# Python Program illustrating> # working of argmax()> > import> numpy as geek> > # Working on 2D array> array> => geek.random.randint(> 16> , size> => (> 4> ,> 4> ))> print> (> 'INPUT ARRAY :
'> , array)> > # No axis mentioned, so works on entire array> print> (> '
Max element : '> , geek.argmax(array))> > # returning Indices of the max element> # as per the indices> > '''> > [[ 0 3 8 13]> > [12 11 2 11]> > [ 5 13 8 3]> > [12 15 3 4]]> > ^ ^ ^ ^> > 12 15 8 13 - element> > 1 3 0 0 - indices> '''> print> (> '
Indices of Max element : '> , geek.argmax(array, axis> => 0> ))> > > '''> > ELEMENT INDEX> > ->[[ 0 3 8 13] 13 3> > ->[12 11 2 11] 12 0> > ->[ 5 13 8 3] 13 1> > ->[12 15 3 4]] 15 1> > '''> print> (> '
Indices of Max element : '> , geek.argmax(array, axis> => 1> ))> |
Sortida:
INPUT ARRAY : [[ 0 3 8 13] [12 11 2 11] [ 5 13 8 3] [12 15 3 4]] Max element : 15 Indices of Max element : [1 3 0 0] Indices of Max element : [3 0 1 1]
Codi 3:
Python
# Python Program illustrating> # working of argmax()> > import> numpy as geek> > # Working on 2D array> array> => geek.arange(> 10> ).reshape(> 2> ,> 5> )> print> (> 'array :
'> , array)> > array[> 0> ][> 1> ]> => 6> print> (> '
array :
'> , array)> > # Returns max element> print> (> '
array : '> , geek.argmax(array))> > # First occurrence of an max element is given> print> (> '
MAX ELEMENT INDICES : '> , geek.argmax(array, axis> => 0> ))> |
Sortida:
array : [[0 1 2 3 4] [5 6 7 8 9]] array : [[0 6 2 3 4] [5 6 7 8 9]] array : 9 MAX ELEMENT INDICES : [1 0 1 1 1]
Nota :
Aquests codis no s'executaran als IDE en línia. Si us plau, executeu-los als vostres sistemes per explorar el funcionament.
