8
C'è un modo per ottenere max e argmax di un colpo?Numpy argmax. Come calcolare sia max che argmax?
import numpy as np
a=[0,0,1,0]
maximum=max(a)
index=np.argmax(a)
Esiste un modo più veloce per farlo, con qualcosa come:
[maximum,index]=function(a)
Buon punto sul risparmio di tempo di calcolo, tuttavia, se la chiamata argmax originale dovesse ordinare una matrice 2D sul secondo asse, la seconda istruzione probabilmente genererebbe un errore. – PaulDong
Puoi recuperarlo usando unravel_index (https://docs.scipy.org/doc/numpy/reference/generated/numpy.unravel_index.html) – PaxRomana99