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

【问题标题】：how can I check if a matrix is square using sympy?如何使用 sympy 检查矩阵是否为正方形？
【发布时间】：2021-05-09 11:14:54
【问题描述】：

当输入是我使用的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。

【问题讨论】：

【解决方案1】：

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

from sympy import shape

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

也可以看看docs

【讨论】：