Recherche rapide Python dans un tableau numpy

Question

Est-ce que Numpy a une fonction de recherche rapide d'élément dans un tableau 2D et retourne ses index? moyenne par exemple:

a=54 
array([[ 0, 1, 2, 3], 
     [ 4, 5, 54, 7], 
     [ 8, 9, 10, 11]]) 

valeur égale sera donc array[1][2]. Bien sûr, je peux le faire en utilisant loops- simple mais je veux quelque chose de similaire à:

if 54 in arr 

Answer 1

In [4]: import numpy as np 

In [5]: my_array = np.array([[ 0, 1, 2, 3], 
          [ 4, 5, 54, 7], 
          [8, 54, 10, 54]]) 

In [6]: my_array 
Out[6]: 
array([[ 0, 1, 2, 3], 
     [ 4, 5, 54, 7], 
     [ 8, 54, 10, 54]]) 

In [7]: np.where(my_array == 54) #indices of all elements equal to 54 
Out[7]: (array([1, 2, 2]), array([2, 1, 3])) #(row_indices, col_indices) 

In [10]: temp = np.where(my_array == 54) 

In [11]: zip(temp[0], temp[1]) # maybe this format is what you want 
Out[11]: [(1, 2), (2, 1), (2, 3)]