numpy判断两个矩阵是否相等

要判断两个NumPy数组或矩阵是否相等，可以使用np.array_equal()函数。该函数会比较两个数组的形状和元素值是否完全一致，如果一致则返回True，否则返回False。

例如，比较两个二维矩阵的代码如下：

import numpy as np

a = np.array([[1, 2], [3, 4]])
b = np.array([[1, 2], [3, 4]])
c = np.array([[1, 2], [4, 3]])

print(np.array_equal(a, b))  # True
print(np.array_equal(a, c))  # False

输出结果为True和False。

另外，如果需要忽略小数点后的误差，可以使用np.allclose()函数。该函数比较两个数组的形状和元素值是否在一定的误差范围内接近，如果一致则返回True，否则返回False。

例如，比较两个浮点数矩阵的代码如下：

import numpy as np

a = np.array([[1.0, 2.0], [3.0, 4.0]])
b = np.array([[1.000001, 2.000002], [2.999999, 3.999998]])

print(np.allclose(a, b, rtol=1e-5, atol=1e-4))  # True

输出结果为True。其中，rtol和atol分别表示相对误差和绝对误差的阈值。