如何检查矩阵中的每一行是否等于数组并返回包含结果的布尔数组？

Question

如何检查矩阵中的每一行是否等于数组并使用NumPy返回包含结果的布尔数组？例如

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

# Expected Result: [False,True,False]

Answer 1

我发现这样做的最好方法是:

result = np.all(a==b, axis=1)