Python矩阵列和行检查

def check_matrix(matrix):
    row_sum = []
    col_sum = []

    # 计算每行的总和
    for row in matrix:
        row_sum.append(sum(row))

    # 计算每列的总和
    for col in range(len(matrix[0])):
        col_sum.append(sum(row[col] for row in matrix))

    return row_sum, col_sum

matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

row_sum, col_sum = check_matrix(matrix)
print("每行总和：", row_sum)
print("每列总和：", col_sum)

每行总和： [6, 15, 24]
每列总和： [12, 15, 18]