web-dev-qa-db-ja.com

Numpy Array行による行インデックス検索を取得します

私はnumpyを初めて使用し、Pythonでランダムフォレストを使用してクラスタリングを実装しています。私の質問は：

配列内の正確な行のインデックスを見つけるにはどうすればよいですか？例えば

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

そして、[0. 0. 3.]を探し、結果1（2行目のインデックス）を取得します。

なにか提案を？コードに従います（動作しません...）

    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
pythonarraysnumpyrandom-forest
17
user2801023

なぜこのようなことをしないのですか？

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