web-dev-qa-db-fra.com

Numpy minimum au format (ligne, colonne)

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!

46
yassin

Utilisation unravel_index :

numpy.unravel_index(A.argmin(), A.shape)
76
Philipp

[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

2
Nikesh Bajaj