Numpy Array Obtenir la recherche d'index de ligne par une ligne

Je suis nouveau sur numpy et je me suis mise en œuvre de clustering avec la forêt au hasard en python. Ma question est:

Comment pourrais-je trouver l'indice de la exacte de la ligne dans un tableau? Par exemple

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

et je regarde pour [0. 0. 3.] et obtenir comme résultat 1(l'index de la deuxième ligne).

Toute suggestion? Suit le code (pas de travail...)

    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

source d'informationauteur user2801023 | 2013-09-20

  1. 40

    Pourquoi ne pas tout simplement faire quelque chose comme cela?

    >>> 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]),)