求2维矩阵(m×n)中每行和每列的和

def calculate_sum(matrix):
    m = len(matrix)  # 矩阵的行数
    n = len(matrix[0])  # 矩阵的列数

    rowSum = [0] * m  # 存储每行的和
    colSum = [0] * n  # 存储每列的和

    # 计算每行的和
    for i in range(m):
        for j in range(n):
            rowSum[i] += matrix[i][j]

    # 计算每列的和
    for j in range(n):
        for i in range(m):
            colSum[j] += matrix[i][j]

    return rowSum, colSum

# 示例矩阵
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

rowSum, colSum = calculate_sum(matrix)

print("每行的和：", rowSum)
print("每列的和：", colSum)

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