检查2D矩阵在python中是否为正方形

Question

任务是检查2D矩阵是否为正方形。我的代码适用于以[[a]，[b]，[c]，[d]]形式测试2d列表，如果是正方形则返回True，否则返回False。但是，当提交给自动测试仪时，它表示期望为true并返回false。

我不知道为什么会这样，将不胜感激。

def is_square(mat):
    i = 0
    while(i < len(mat)):
        if(len(mat[i]) != len(mat)):
            return False
        i += 1
    return True

Answer 1

如果迭代的所有元素均为all，则函数true返回true。

def is_square(mat):
    return all([len(i)==len(mat) for i in mat])