V mnohých prípadoch, keď je veľkosť poľa príliš veľká, trvá príliš dlho, kým z nich nájde maximum prvkov. Na tento účel poskytuje numpy modul Pythonu funkciu tzv numpy.argmax() . Táto funkcia vracia indexy maximálnych hodnôt spolu so zadanou osou.

Syntax:

 numpy.argmax(a, axis=None, out=None) 

Parametre

x: pole_like

Tento parameter definuje zdrojové pole, ktorého maximálnu hodnotu chceme poznať.

reverzný reťazec java

os: int (voliteľné)

Tento parameter definuje os, pozdĺž ktorej je index prítomný, a štandardne je do splošteného poľa.

out: pole (voliteľné)

Tento parameter definuje ndarray, do ktorého sa vloží výsledok. Bude rovnakého typu a tvaru, ktorý je vhodný na uloženie výsledku

Návraty

Tento parameter definuje ndarray, ktorý obsahuje indexy poľa. Tvar je rovnaký ako x.tvar s odstráneným rozmerom pozdĺž osi.

Príklad 1:

 Import numpy as np x = np.arange(20).reshape(4,5) + 7 x y=np.argmax(a) y 

Výkon:

 array([[ 7, 8, 9, 10, 11], [12, 13, 14, 15, 16], [17, 18, 19, 20, 21], [22, 23, 24, 25, 26]]) 19 

Vo vyššie uvedenom kóde

nerovná sa mysql

• Importovali sme numpy s aliasom np.
• Vytvorili sme pole 'X' použitím np.arange() funkciu s tvarom štyroch riadkov a piatich stĺpcov.
• Do každého prvku poľa sme pridali aj 7.
• Deklarovali sme premennú 'a' a priradil vrátenú hodnotu np.argmax() funkciu.
• Prešli sme pole 'X' vo funkcii.
• Nakoniec sme sa pokúsili vytlačiť hodnotu 'a' .

Vo výstupe zobrazuje indexy maximálneho prvku v poli.

Príklad 2:

 Import numpy as np x = np.arange(20).reshape(4,5) + 7 y=np.argmax(x, axis=0) z=np.argmax(x, axis=1) y z 

Výkon:

 array([3, 3, 3, 3, 3], dtype=int64) array([4, 4, 4, 4], dtype=int64) 

Príklad 3:

 Import numpy as np x = np.arange(20).reshape(4,5) + 7 indices = np.unravel_index(np.argmax(x, axis=None), x.shape) indices x[indices] 

Výkon:

 (3, 4) 26 

Príklad 4:

 import numpy as np a = np.array([[5,2,1], [3,7,9],[0, 4, 6]]) index_arr = np.argmax(a, axis=-1) index_arr # Same as np.max(a, axis=-1, keepdims=True) result = np.take_along_axis(a, np.expand_dims(index_arr, axis=-1), axis=-1) result1 # Same as np.max(a, axis=-1) result = np.take_along_axis(a, np.expand_dims(index_arr, axis=-1), axis=-1).squeeze(axis=-1) result2 

Výkon:

 array([[0], [2], [2]]) array([5, 9, 6]) 

Vo vyššie uvedenom kóde

bash inak ak

• Importovali sme numpy s aliasom np.
• Vytvorili sme viacrozmerné pole 'a ' použitím np.array() funkciu.
• Deklarovali sme premennú 'index_arr' a priradil vrátenú hodnotu np.argmax() funkciu.
• Prešli sme pole 'a' a os vo funkcii.
• Pokúsili sme sa vytlačiť hodnotu 'index_arr' .
• Nakoniec sme sa pokúsili získať maximálnu hodnotu poľa pomocou dvoch rôznych spôsobov, ktoré sú dosť podobné np.argmax() .

Vo výstupe zobrazuje indexy maximálnych prvkov v poli a hodnoty, ktoré sú na týchto indexoch prítomné.