python - 判断一个矩阵是否为单位矩阵(numpy)

Question

确定给定矩阵“M”是否正确的最佳方法是什么等于身份？ IE。像这样的东西:

if numpy.identity(3) == M:
     ...

Answer 1

np.eye 或 np.identity 都将返回指定大小的单位矩阵 I。例如:

np.eye(3) # np.identity(3)
array([[ 1.,  0.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  0.,  1.]])

假设 M 是正方形并且 dtype=int，这就是您要测试的方式:

assert (M.shape[0] == M.shape[1]) and (M == np.eye(M.shape[0])).all()

添加检查以确保 M 首先是正方形。如果两个二维数组相同，则返回 True。

或者，如果 M 是浮点矩阵，请改用 np.allclose:

assert (M.shape[0] == M.shape[1]) and np.allclose(M, np.eye(M.shape[0]))