Wie erhält man Zeilen in einem Pandas-Datenrahmen mit Maximalwerten in einer Spalte und behält den ursprünglichen Index bei?

numpy: Wie bekomme ich ein Maximum aus einem Argmax-Ergebnis?

Ich habe ein numpy Array von beliebiger Form, a = array([[[ 1, 2], [ 3, 4], [ 8, 6]], [[ 7, 8], [ 9, 8], [ 3, 12]]]) a.shape = (2, 3, 2)und ein Ergebnis von argmax über der letzten Achse: np.argmax(a, axis=-1) = array([[1, 1, 0], [1, 0, 1]]) ...