确保矩阵行的长度相同（python3）

Question

所以我有这个 python 3 代码来输入一个矩阵：

matrix = []
lop=True
while lop:
    line = input()
    if not line:
        lop=False
    if matrix != []:
        if len(line.split()) != len(matrix[-1]):
            print("Not same length")
            menu()
    values = line.split()
    row = [int(value) for value in values]
    matrix.append(row)

但是，如果我输入

1 2 3
4 5 6 7
8 9 0 1 2

我的代码会让它通过，但您会注意到第 2 行和第 3 行的长度与第 1 行不同；如何防止？该行必须与第 1 行长度相同，否则它必须返回一条错误消息，例如“行的长度不同。我不太确定该怎么做。也许：

for row in matrix:
    if len(row) == matrix[1]
        pass
    else:
       print('not same length')

但它不起作用。

谢谢

Answer 1

len(set(map(len,matrix))) == 1

解释：

map(len,matrix) 产生矩阵所有行的长度

set(...) 给出存在的所有唯一/不同长度的行。

• 如果所有行的长度相同，这将是一个包含 1 个单个元素的集合。
• 否则，它将有 2 个或更多。

最后，len(...) == 1 返回我们上面得到的是否包含 1 个单个元素，即所有行是否具有相同的长度。

Answer 2

如果你想匹配第一行的长度，试试这个方法，

使用len(matrix[0])

for row in matrix:
    if len(row) == len(matrix[0]):
        pass
    else:
       print('not same lenght')

Answer 3

使用内置的len() 函数和break 语句。

matrix = []
lop =True
while lop:
    line = input('Enter your line: ')
    if not line:
        lop=False
    if matrix != []:
        if len(line.split()) != len(matrix[-1]):
            print("Not same length")
            break
    values = line.split()
    row = [int(value) for value in values]
    matrix.append(row)

运行如下：

bash-3.2$ python3 matrix.py
Enter your line: 1 2 3
Enter your line: 4 5 6
Enter your line: 7 8 9 0
Not same length
bash-3.2$