检查矩阵是否为正方形[重复]

Question

这个问题已经在这里有了答案：

• 检查矩阵是否为正方形？ （Python） 5个答案

嘿，我是python新手，只是想知道，在提交之前我做的正确吗？

inverse = []
matrix = [[1, 2] , [3, 4]]
if len (matrix) != len(matrix [0]):
raised (‘The matrix must be square’)

Answer 1

您需要引发错误类型并更改逻辑以确定矩阵是否为正方形：

inverse = []
matrix = [[1, 2], [3, 4]]
if len(matrix) != len(matrix[0]) and len(set(map(len, matrix))) != 1:
   raise AttributeError('The matrix must be square') #or error of your choice

但是，如果您希望生成自定义错误，则可以继承Exception并覆盖__str__ ：

class MatrixLenError(Exception):
    pass

matrix = [[1, 2], [3, 4]]
if len(matrix) != len(matrix[0]) and len(set(map(len, matrix))) != 1:
    raise MatrixLenError('The matrix must be square')

Answer 2

尽管您的代码有效，但不适用于[[1,2], [1,2,3]]这样的矩阵。

首先，将此逻辑封装在一个方法中：

def is_squared(matrix):
    # Check that all rows have the correct length, not just the first one
    return all(len(row) == len(matrix) for row in matrix)

然后，您可以：

inverse = []
matrix = [[1,2], [3,4]]
if not is_squared(matrix):
    raise AttributeError("The matrix must be squared")