python 行列から完全一致の一行を取り出したいです

Question

前提・実現したいこと

array = [[x1 y1 z1] [x2 y2 z2] ... [xn yn zn]]から、
全ての成分がp = [x y z]と一致する行だけを抽出したいです。

同じものが存在しない可能性は考慮する必要はありません。

発生している問題

np.where(array == p)で抽出すると、
「3成分のうち、少なくとも1成分が一致した行」を全て抽出してくるようです。

良い改善策はないでしょうか。
よろしくお願いします。

Answer 1

このサイトを参考にして以下のコード

import numpy as np

array = np.array([[1,2,3],[4,5,6],[7,8,9],[0,1,2],[3,4,5]])
p = np.array([0,1,2])

for data in array:
    if np.allclose(data,p):
        result = data

print(result)

Comments

[Sayuki]

ありがとうございます。
自分はpython初心者なのでallclose教えていただいて勉強になりました。

Answer 2

.all(axis=1)を使うのはどうでしょう。

array = np.array([
    [1,2,3],
    [4,5,6],
    [7,8,9],
    [0,1,2],
    [3,4,5]
])
p = np.array([0,1,2])

match = (array == p).all(axis=1)
index = np.where(match)

Find matching rows in 2 dimensional numpy array

Comments

[Sayuki]

ありがとうございます。
allに関して調べましたが、なかなかこういう使い方が見つけられなかったので本当に助かりました。