检查一个二维矩阵在蟒蛇方

Question

任务是检查是否二维矩阵是方形的。我的代码工作与形式测试二维名单[A]，[B]，[C]，[D]]，返回true，如果它是方形和False如果不是。然而，提交到自动测试仪时，它说，它预计真正的，并得到返回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])