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

当输入类型为

np.ndarray

i时，检查矩阵是否为正方形

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。

shape函数应该执行您想要的操作：

from sympy import shape

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

---

还可以看看

还有

M.is\u square

。