Wie finde ich die Zeile oder Spalte, die den Array-weiten Maximalwert in einem 2d-numpy-Array enthält?
Wenn Sie nur das eine oder das andere brauchen:
für die Spalte und
für die Zeile.
Sie können np.where(x == np.max(x)) verwenden.
np.where(x == np.max(x))
Zum Beispiel:
Der erste Wert ist die Zeilennummer, die zweite Nummer ist die Spaltennummer.
Sie können np.argmax zusammen mit np.unravel_index wie in
np.argmax
np.unravel_index
np.argmax gibt nur den Index des (ersten) größten Elements im abgeflachten Array zurück. Wenn Sie also die Form Ihres Arrays kennen (was Sie tun), können Sie die Zeilen- / Spaltenindizes leicht finden:
Tags und Links python numpy