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

Question

当输入类型为np.ndarrayi时，检查矩阵是否为正方形

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

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

Answer 1

shape函数应满足您的要求：

from sympy import shape

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

还可以看看docs