比较两个NumPy数组是否相等，按元素

比较两个NumPy数组是否相等，按元素

(A==B).all()

测试数组（A == B）的所有值是否均为True。

注意：也许您还想测试A和B形状，例如

A.shape == B.shape

特殊情况和替代方法 （来自dbaupp的回答和yoavram的评论）

应当指出的是：

• 在特定情况下，该解决方案可能会产生奇怪的行为：如果

  A

  或

  B

  为空，而另一个包含单个元素，则返回

  True

  。由于某种原因，比较会

  A==B

  返回一个空数组，

  all

  运算符将为此返回一个空数组

  True

  。
• 另一个风险是，如果

  A

  且

  B

  形状不相同且不可广播，则此方法将引发错误。

总之，如果你有一个关于怀疑

A

和

B

形状或只是想安全：的专业功能用途之一：

np.array_equal(A,B)  # test if same shape, same elements valuesnp.array_equiv(A,B)  # test if broadcastable shape, same elements valuesnp.allclose(A,B,...) # test if same shape, elements have close enough values