Numpy np.argmax
Numpy ti fornisce la funzione argmax () che consente di recuperare l'indice dell'elemento massimo all'interno di un array su un asse specifico.
Questo tutorial tenta di spiegare come il numpy.La funzione argmax () funziona e come usarla.
Sintassi della funzione
La sintassi della funzione è come mostrato di seguito:
numpy.ArgMax (array, axis = nessuno, out = nessuno, *, KeepDims =La funzione accetta i seguenti parametri:
- array - si riferisce all'array di input su cui viene applicata la funzione.
- Asse -Specifica lungo il quale viene applicato l'asse dell'array la funzione: questo è un parametro di tipo intero opzionale. La funzione appiattirà l'array se il valore dell'asse non è impostato.
- Out - Specifica un array in cui viene inserito il risultato. L'array specificato dovrebbe essere del tipo e della forma adatti.
- KeepDims - Questo è un parametro booleano che richiede vero o falso. Se impostato su true, l'asse ridotto viene lasciato come dimensioni con una dimensione.
Valore di ritorno della funzione
La funzione restituisce un array di indici nell'array con la stessa forma dell'array.forma.
Esempio
Illustriamo come usare il numpy.Argmax funzione con un esempio.
Inizia importando numpy come:
# importy numpyImporta Numpy come NP
Quindi, crea un array 2D come mostrato:
# Array 2Darray = np.Arange (6).Reshape ((2,3))
Stampa (array)
Quanto sopra dovrebbe creare un array 2D con gli elementi come mostrato:
[[0 1 2][3 4 5]]
Per utilizzare la funzione argmax () sull'intero array (appiattito), possiamo eseguire il codice come mostrato:
# elemento maxprint (f "max value: np.argmax (array, axis = nessuno) ");
Il codice sopra restituisce l'indice massimo del valore massimo come mostrato:
Puoi anche usare la funzione ArgMax lungo un asse specifico come mostrato:
# elemento max lungo 0 assiStampa (NP.argmax (array, axis = 0));
Nota: poiché stiamo lavorando con un array 2D, l'array ha due possibili assi. Il primo va in verticale attraverso le righe come asse 0 e l'altro corre orizzontalmente attraverso le colonne come asse 1.
Quindi, il codice sopra dovrebbe restituire:
Lo stesso caso si applica lungo l'asse 1. Un esempio è come mostrato:
# elemento max lungo l'asse 1Stampa (NP.argmax (array, axis = 1))
Questo dovrebbe tornare:
[2 2]Esempio 2
Possiamo anche usare la funzione ArgMax con un array N-dimensionale. Ad esempio, il codice seguente illustra come utilizzare la funzione con un array 3D.
# Array 3Darray = np.Arange (24).Reshape (2, 3, 4)
Stampa (array)
Questo dovrebbe generare un array come:
[[[0 1 2 3][4 5 6 7]
[8 9 10 11]]
[[12 13 14 15]
[16 17 18 19]
[20 21 22 23]]
Lungo gli assi, possiamo ottenere l'argmax come mostrato:
# elemento maxprint (f "max: np.argmax (array, axis = nessuno) ")
# elemento max lungo l'asse 0
print (f "max indici: np.argmax (array, axis = 0) ")
# elemento max lungo l'asse 1
print (f "max indici: np.argmax (array, axis = 1) ")
# elemento max lungo l'asse 2
print (f "max indici: np.argmax (array, axis = 2) ")
Il codice sopra dovrebbe restituire:
Max: 23Indici massimi: [[1 1 1 1]
[1 1 1 1]
[1 1 1 1]]
Indici massimi: [[2 2 2 2]
[2 2 2 2]]
Indici massimi: [[3 3 3]
[3 3 3]]
Chiusura
Questo articolo esplora come utilizzare la funzione numpy argmax () per recuperare gli indici dei valori massimi insieme a un indice specifico.
Grazie per aver letto e rimanete sintonizzati!!