logo

numpy.argmax() v Pythonu

The numpy.argmax() funkcija vrne indekse največjega elementa matrike na določeni osi.

Sintaksa:



java filter tok
numpy.argmax(array, axis = None, out = None)>

Parametri:

 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>

Povratek:

Array of indices into the array with same shape as array.shape with the dimension along axis removed.>

Koda 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>))>

>

>

Izhod:

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]>

Koda 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>))>

pretvori niz v enum

>

>

Izhod:

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]>

Koda 3:

Python


tipkopis foreach



# 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>))>

>

>

Izhod:

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]>

Opomba :
Te kode se ne izvajajo v spletnih IDE. Zaženite jih v svojih sistemih, da raziščete delovanje.