Comment puis-je connaître l'indice (ligne, colonne) du minimum d'un tableau/matrice numpy?
Par exemple, si A = array([[1, 2], [3, 0]])
, je veux obtenir (1, 1)
Merci!
Utilisation unravel_index
:
numpy.unravel_index(A.argmin(), A.shape)
[Faute de frappe corrigée]
Une autre solution simple est
ri, ci = A.argmin()//A.shape[1], A.argmin()%A.shape[1]
Comme numpy.argmin retourne la lecture de l'index dans l'ordre des lignes principales
Oui, vous avez raison, c'était une faute de frappe, qui fonctionnait pour la matrice carrée