Coordonnées de l'élément sur le tableau numpy

J'ai un tableau numpy:

[[  0.   1.   2.   3.   4.]
 [  7.   8.   9.  10.   4.]
 [ 14.  15.  16.  17.   4.]
 [  1.  20.  21.  22.  23.]
 [ 27.  28.   1.  20.  29.]]

qui je veux trouver rapidement les coordonnées des valeurs spécifiques et éviter python boucle sur le tableau. Par exemple le nombre 4 est sur:

row 0 and col 4
row 1 and col 4
row 2 and col 4

et une fonction de recherche retourne un tuple:

((0,4),(1,4),(2,4))

Cela peut être fait directement via nunmpy fonctions?

source d'informationauteur | 2011-03-30

  1. 18

    Si a est votre tableau, vous pouvez utiliser:

    ii = np.nonzero(a == 4)

    ou 

    ii = np.where(a == 4)

    Si vous voulez vraiment un tuple, vous pouvez convertir le n-uplet de tableaux pour le tuple de tuples, mais la valeur de retour de fonctions de numpy est convient pour ensuite faire d'autres opérations sur votre tableau.

    Conversion d'un tuple dans le cas des OP spécifications:

    tuple(zip(*ii))
  2. 11
    a = numpy.array([[  0.,  1.,  2.,  3.,  4.],
                 [  7.,  8.,  9., 10.,  4.],
                 [ 14., 15., 16., 17.,  4.],
                 [  1., 20., 21., 22., 23.],
                 [ 27., 28.,  1., 20., 29.]])
print numpy.argwhere(a == 4.)

    imprime

    [[0 4]
 [1 4]
 [2 4]]

    Les mises en garde habituelles pour virgule flottante comparaisons appliquer.