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

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

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

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

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

解决方法

形状函数应该做你想做的：

from sympy import shape

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

另请看docs