Numpy NP.Argmax
Numpy gir deg ArgMax () -funksjonen som lar deg hente indeksen for MAX -elementet i en matrise på en bestemt akse.
Denne opplæringen prøver å forklare hvordan Numpy.ArgMax () -funksjon fungerer og hvordan du bruker den.
Funksjonssyntaks
Syntaksen til funksjonen er som vist nedenfor:
numpy.Argmax (matrise, akse = ingen, ut = ingen, *, keepDims =Funksjonen godtar følgende parametere:
- Array - refererer til inngangsarrayen som funksjonen brukes.
- Axis -Spesificifices Likre hvilken akse av matrisen funksjonen blir brukt -dette er en valgfritt heltallstypeparameter. Funksjonen vil flate ut matrisen hvis aksen ikke er angitt.
- Ut - spesifiserer en rekke som resultatet settes inn i. Den spesifiserte matrisen skal være av passende type og form.
- Keepdims - Dette er en boolsk parameter som tar enten sant eller usant. Hvis den er satt til sann, blir den reduserte aksen igjen som dimensjoner med størrelse en.
Funksjonsretur
Funksjonen returnerer en rekke indekser i matrisen med samme form som matrisen.form.
Eksempel
La oss illustrere hvordan vi bruker Numpy.Argmax -funksjon med et eksempel.
Begynn med å importere Numpy som:
# importy numpyImporter numpy som NP
Neste, lag en 2D -matrise som vist:
# 2D -matriseArray = NP.Arange (6).omforming ((2,3)))
Print (Array)
Ovennevnte skal lage en 2D -matrise med elementene som vist:
[[0 1 2][3 4 5]]
For å bruke ArgMax () -funksjonen på hele arrayen (flatet), kan vi kjøre koden som vist:
# MAX ElementPrint (F "Maks verdi: NP.argmax (matrise, akse = ingen) ");
Koden over returnerer maks -indeksen for maks. Som vist:
Du kan også bruke Argmax -funksjonen langs en spesifikk akse som vist:
# Maks element langs 0 -aksentrykk (NP.argmax (matrise, akse = 0));
Merk: Siden vi jobber med en 2D -matrise, har matrisen to mulige akser. Den første går vertikalt over radene som akse 0, og den andre løper horisontalt over kolonner som akse 1.
Derfor bør koden ovenfor komme tilbake:
Det samme tilfellet gjelder langs akse 1. Et eksempel er som vist:
# Maks element langs akse 1trykk (NP.Argmax (Array, Axis = 1))
Dette skulle komme tilbake:
[2 2]Eksempel 2
Vi kan også bruke Argmax-funksjonen med en N-dimensjonsarray. For eksempel illustrerer koden nedenfor hvordan du bruker funksjonen med en 3-D-matrise.
# 3D -matriseArray = NP.Arange (24).omforme (2, 3, 4)
Print (Array)
Dette skal generere en matrise som:
[[0 1 2 3][4 5 6 7]
[8 9 10 11]]
[[12 13 14 15]
[16 17 18 19]
[20 21 22 23]]]
Langs aksene kan vi få Argmax som vist:
# MAX Elementtrykk (f "maks: np.argmax (matrise, akse = ingen) ")
# Maks element langs akse 0
Print (F "Maks indekser: NP.Argmax (matrise, akse = 0) ")
# Maks element langs akse 1
Print (F "Maks indekser: NP.Argmax (matrise, akse = 1) ")
# Maks element langs akse 2
Print (F "Maks indekser: NP.ArgMax (Array, Axis = 2) ")
Koden over skal returnere:
Maks: 23Maks indekser: [[1 1 1 1]
[1 1 1 1]
[1 1 1 1]]
Maks indekser: [[2 2 2 2]
[2 2 2 2]]
Maks indekser: [[3 3 3]
[3 3 3]]
Lukking
Denne artikkelen undersøker hvordan du bruker Numpy ArgMax () -funksjonen for å hente indeksene for maksimalverdiene sammen med en spesifikk indeks.
Takk for at du leser, og følg med!!