Python의 numpy.argmax()
그만큼 numpy.argmax() 함수는 특정 축에서 배열의 최대 요소 인덱스를 반환합니다.
구문:
numpy.argmax(array, axis = None, out = None)
매개변수:
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
반품 :
Array of indices into the array with same shape as array.shape with the dimension along axis removed.
코드 1:
파이썬
# 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> ))> |
출력 :
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]
코드 2:
파이썬
# 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> ))> |
출력 :
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]
코드 3:
파이썬
# 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> ))> |
출력 :
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]
메모 :
이 코드는 온라인 IDE에서 실행되지 않습니다. 작업을 탐색하려면 시스템에서 실행하십시오.
