Array Numpy Ricerca riga di indice per riga

Question

Sono nuovo in numpy e sto implementando il clustering con una foresta casuale in python. La mia domanda è:

Come posso trovare l'indice della riga esatta in una matrice? Ad esempio

[[ 0. 5. 2.] 
[ 0. 0. 3.] 
[ 0. 0. 0.]] 

e cerco [0. 0. 3.] e ottenere come risultato 1 (l'indice della seconda riga).

Qualche suggerimento? Segue il codice (non funziona ...)

for index, element in enumerate(leaf_node.x): 
     for index_second_element, element_two in enumerate(leaf_node.x): 
      if (index <= index_second_element): 
       index_row = np.where(X == element) 
       index_column = np.where(X == element_two) 
       self.similarity_matrix[index_row][index_column] += 1 

Answer 1

Perché non fare semplicemente qualcosa del genere?

>>> a 
array([[ 0., 5., 2.], 
     [ 0., 0., 3.], 
     [ 0., 0., 0.]]) 
>>> b 
array([ 0., 0., 3.]) 

>>> a==b 
array([[ True, False, False], 
     [ True, True, True], 
     [ True, True, False]], dtype=bool) 

>>> np.all(a==b,axis=1) 
array([False, True, False], dtype=bool) 

>>> np.where(np.all(a==b,axis=1)) 
(array([1]),)