python - 如何使用 sympy 检查矩阵是否为正方形？

Question

当输入是np.ndarray我使用的类型时检查矩阵是否为正方形

if len(V) != len(V[0]):
        raise ValueError('V is not a square matrix')

但是当我想使用 sympy 时len(V)，例如对于 2x2 矩阵，返回 4 而不是 2。我尝试使用v.shapewhich 返回 (2,2)，但如果输入矩阵，我不知道如何实现 ValueError与此输出不相符。

标签： pythonpython-3.xmatrixsympy

Answer 1

形状功能应该做你想做的事：

from sympy import shape

def isSquare(M):
   s = shape(M)
   return s[0] == s[1]

也看看文档