检查矩阵是否为正方形？（蟒蛇）

Question

我想测试[[5,6]，[7,8]]的2x2矩阵，看看它是否是正方形。

我运行我的代码，我应该得到True，但是我得到False ...

 def square(sq):
     for element in sq:
         if element:
             return False
         return True

Answer 1

给定的m是一个numpy矩阵并且您已经导入numpy

def square(m):
    return m.shape[0] == m.shape[1]

Answer 2

如果要检查矩阵是否为NxN，可以使用：

def isSquare (m): return all (len (row) == len (m) for row in m)

如您在评论中所说：如果所有行的长度等于行数。