Numpy NP.Argmax
Numpy bietet Ihnen die Funktion argmax (), mit der Sie den Index des maximalen Elements in einem Array auf einer bestimmten Achse abrufen können.
Dieses Tutorial versucht zu erklären, wie der Numpy.Argmax () -Funktion funktioniert und wie man sie benutzt.
Funktionssyntax
Die Syntax der Funktion ist wie unten gezeigt:
Numpy.Argmax (Array, Axis = Keine, out = keine, *, keepdims =Die Funktion akzeptiert die folgenden Parameter:
- Array - bezieht sich auf das Eingangsarray, auf das die Funktion angewendet wird.
- Axis -spezifiziert entlang der Achse des Arrays Die Funktion wird angewendet. Dies ist ein optionaler Parameter vom Typ ganzzahliger Typ. Die Funktion wird das Array abgeflacht, wenn der Achsenwert nicht festgelegt ist.
- Out - Gibt ein Array an, in das das Ergebnis eingefügt wird. Das angegebene Array sollte von der geeigneten Art und Form sein.
- Keepdims - Dies ist ein boolescher Parameter, der entweder wahr oder falsch ist. Wenn auf True gesetzt, bleibt die reduzierte Achse als Abmessungen mit der ersten Größe.
Funktionsrückgabewert
Die Funktion gibt ein Array von Indizes mit der gleichen Form wie das Array in das Array zurück.Form.
Beispiel
Lassen Sie uns veranschaulichen, wie man den Numpy benutzt.Argmax -Funktion mit einem Beispiel.
Importieren Sie zunächst Numpy als:
# Importy NumpyNumph als NP importieren
Erstellen Sie als nächstes ein 2D -Array wie gezeigt:
# 2D -ArrayArray = np.Arange (6).Umformung ((2,3))
Druck (Array)
Das obige sollte ein 2D -Array mit den gezeigten Elementen erstellen:
[[0 1 2][3 4 5]]
Um die Funktion argmax () im gesamten Array (abgeflacht) zu verwenden, können wir den Code wie gezeigt ausführen:
# Max -Elementprint (f "max value: np.argmax (array, axis = keine) ");
Der obige Code gibt den maximalen Index des maximalen Wertes wie gezeigt zurück:
Sie können auch die Argmax -Funktion entlang einer bestimmten Achse wie gezeigt verwenden:
# maximales Element entlang der 0 Achsedrucken (np.argmax (Array, axis = 0));
Hinweis: Da wir mit einem 2D -Array arbeiten, hat das Array zwei mögliche Achsen. Die erste gehen vertikal über die Reihen als Achse 0 und die anderen läuft horizontal über Säulen als Achse 1.
Daher sollte der obige Code zurückgeben:
Der gleiche Fall gilt entlang der Achse 1. Ein Beispiel ist wie gezeigt:
# maximales Element entlang der Achse 1drucken (np.Argmax (Array, Achse = 1))
Dies sollte zurückkehren:
[2 2]Beispiel 2
Wir können die Argmax-Funktion auch mit einem n-dimensionsartigen Array verwenden. Der folgende Code zeigt beispielsweise, wie die Funktion mit einem 3-D-Array verwendet wird.
# 3D -ArrayArray = np.Arange (24).Umform (2, 3, 4)
Druck (Array)
Dies sollte ein Array erzeugen wie:
[[0 1 2 3][4 5 6 7]
[8 9 10 11]]
[[12 13 14 15]
[16 17 18 19]
[20 21 22 23]]]
Entlang der Achsen können wir den Argmax wie gezeigt bekommen:
# Max -Elementprint (f "max: np.Argmax (Array, Axis = Keine) ")
# Max -Element entlang der Achse 0
print (f "Max -Indizes: np.Argmax (Array, Axis = 0) ")
# maximales Element entlang der Achse 1
print (f "Max -Indizes: np.Argmax (Array, Axis = 1) ")
# Max -Element entlang der Achse 2
print (f "Max -Indizes: np.Argmax (Array, Axis = 2) ")
Der obige Code sollte zurückgeben:
Max: 23Maximale Indizes: [[1 1 1 1]
[1 1 1 1]
[1 1 1 1]]
Maximale Indizes: [[2 2 2 2]
[2 2 2 2]]
Max -Indizes: [[3 3 3]
[3 3 3]]
Schließen
In diesem Artikel wird untersucht.
Vielen Dank für das Lesen und bleiben Sie dran!!