Python numpy.argmax()

Suraj Joshi 30 Januar 2023 14 November 2020 NumPy
  1. Syntax von numpy.argmax():
  2. Beispielcodes: numpy.argmax() Methode zum Finden von Indizes mit den größten Werten in einem Array
  3. Beispiel-Codes: Setzen Sie den Parameter axis in der Methode numpy.argmax(), um Indizes der größten Werte in einem Array zu finden
  4. Beispiel-Codes: Setzen Sie den Parameter out in der Methode numpy.argmax(), um Indizes der größten Werte in einem Array zu finden
Python numpy.argmax()

Python Numpy numpy.argmax() gibt die Indizes der Werte mit den höchsten Werten im angegebenen NumPy-Array zurück.

Syntax von numpy.argmax():

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

Parameter

a Array oder Objekt, das in ein Array konvertiert werden kann, in dem wir Indizes mit den höchsten Werten finden müssen.
axis den Index des größten Wertes entlang der Zeile (axis=0) oder Spalte (axis=1) finden. Standardmäßig wird der Index des größten Wertes durch Verflachen des Arrays gefunden.
out Platzhalter für das Ergebnis der np.argmax Methode. Er muss die geeignete Größe haben, um das Ergebnis zu halten.

Zurück

Ein Array von Indizes der Elemente mit dem höchsten Wert im gesamten Array.

Beispielcodes: numpy.argmax() Methode zum Finden von Indizes mit den größten Werten in einem Array

Finden Sie den Index des höchsten Wertes in einem 1-D-Array

import numpy as np

a=np.array([2,6,1,5])

print("Array:")
print(a)

req_index=np.argmax(a)
print("\nIndex with the largest value:")
print(req_index)

print("\nThe largest value in the array:")
print(a[req_index])

Ausgabe:

Array:
[2 6 1 5]

Index with the largest value:
1

The largest value in the array:
6

Sie gibt den Index des Elements mit dem größten Wert im gegebenen Eingabefeld aus.

Wir können den größten Wert im Array auch mit dem Index finden, der von der np.argmax() Methode zurückgegeben wird.

Wenn wir zwei größte Werte im Array haben, gibt die Methode den Index des größten Elements zurück, das zuerst im Array erscheint.

import numpy as np

a=np.array([2,6,1,6])

print("Array:")
print(a)

req_index=np.argmax(a)
print("\nIndex with the largest value:")
print(req_index)

print("\nThe largest value in the array:")
print(a[req_index])

Ausgabe:

Array:
[2 6 1 5]

Index with the largest value:
1

The largest value in the array:
6

Hier ist 6 der höchste Wert, der zweimal im Array vorkommt, aber die Methode np.argmax() gibt den Index von 6 bei Index 1 zurück, da er zuerst im Array vorkommt.

Den Index des höchsten Wertes in einem 2D-Array finden

import numpy as np

a=np.array([[2,1,6],
            [7,4,5]])

print("Array:")
print(a)

req_index=np.argmax(a)
print("\nIndex with the largest value:")
print(req_index)

Ausgabe:

Array:
[[2 1 6]
 [7 4 5]]

Index with the largest value:
3

Da hier der Parameter axis nicht angegeben ist, wird das Array abgeflacht und der Index des größten Elements im abgeflachten Array zurückgegeben.

Beispiel-Codes: Setzen Sie den Parameter axis in der Methode numpy.argmax(), um Indizes der größten Werte in einem Array zu finden

Indizes des höchsten Elements entlang der Spaltenachse finden

import numpy as np

a=np.array([[2,1,6],
            [7,4,5]])

print("Array:")
print(a)

req_index=np.argmax(a,axis=0)
print("\nIndices with the largest value along column axis:")
print(req_index)

Ausgabe:

Array:
[[2 1 6]
 [7 4 5]]

Index with the largest value:
[1 1 0]

Wenn wir axis=0 setzen, ergibt dies Indizes mit dem höchsten Wert entlang jeder Spalte. Hier hat die erste Spalte den höchsten Wert bei Index 1, die zweite Spalte hat den höchsten Wert bei Index 1 und die dritte Spalte hat den höchsten Wert bei Index 0.

Indizes der höchsten Elemente entlang der Zeilenachse finden

import numpy as np

a=np.array([[2,1,6],
            [7,4,5]])

print("Array:")
print(a)

req_index=np.argmax(a,axis=1)
print("\nIndices with the largest value along row axis:")
print(req_index)

Ausgabe:

Array:
[[2 1 6]
 [7 4 5]]

Indices with the largest value along the row axis:
[2 0]

Wenn wir axis=1 setzen, gibt sie die Indizes der höchsten Werte entlang jeder Zeile aus.

Beispiel-Codes: Setzen Sie den Parameter out in der Methode numpy.argmax(), um Indizes der größten Werte in einem Array zu finden

import numpy as np

a=np.array([[2,1,6],
            [7,4,5]])

req_index=np.array(0)

print("Array:")
print(a)

np.argmax(a,out=req_index)
print("\nIndex with the largest value:")
print(req_index)

Ausgabe:

Array:
[[2 1 6]
 [7 4 5]]

Index with the largest value:
3

Hier fungiert die Variable req_index als Platzhalter für den erforderlichen Wert des Index, und wir müssen sicherstellen, dass die Größe der Ausgabe gleich req_index sein muss.

Author: Suraj Joshi
Suraj Joshi avatar Suraj Joshi avatar

Suraj Joshi is a backend software engineer at Matrice.ai.

LinkedIn