如果两个矩阵相同,则仅返回一个布尔语句

时间:2015-12-12 15:41:11

标签: python python-2.7 numpy matrix

假设有一个矩阵A和一个矩阵B。是否存在一个逻辑语句,根据True的所有元素是否与False中的所有元素相同,只能返回AB一个值?

例如A = array([[1, 0, 0],[0, 1, 0]])B = array([[1, 0, 0],[0, 1, 0]])A == B每行和每列的每个元素都会返回TrueFalse

1 个答案:

答案 0 :(得分:3)

使用np.array_equal

此外,您可以通过比较.all()A==B应用于您获得的equality-bool-array,如下所示:

(A==B).all()

后者的效率略低于前者(创建一个临时的bool数组),但同样普遍。

如果比较浮点数,您通常希望值接近但不一定相同,请使用np.allclose