问题描述
我有两个 NumPy ndarrays a
和 b
。我想检查它们是否等于 a=b.copy()
。但是,如果我使用 ==
运算符,它会返回 True
或 False
的元素列表。我想要一个二元答案,即如果所有元素在值、类型等方面都相等,则应返回单个 True
否则 False
。 NumPy 中是否有运算符可以实现此目的?
解决方法
您正在寻找this:
np.array_equal(a,b)
或者:
(a==b).all()
此外,根据您的应用程序和数组数据类型,我建议您也检查 np.allclose
和 np.array_equiv
。