📜  Python|不均匀大小列表中的列总和(1)

📅  最后修改于: 2023-12-03 15:19:17.373000             🧑  作者: Mango

Python | 不均匀大小列表中的列总和

当处理不均匀大小的列表时,可能需要计算每一列的总和。在Python中,我们可以使用zip()函数将列表转换为元组列表,并使用for循环遍历元组列表来计算每一列的总和。

下面是一个示例代码:

def column_sum(matrix):
    # 转置矩阵
    transpose = zip(*matrix)
    # 计算每一列的总和
    sum_list = [sum(column) for column in transpose]
    return sum_list

使用上述代码,我们可以将矩阵转换为元组列表,并在for循环中计算每一列的总和。将每一列的总和存储在sum_list中并进行返回。以下是一个用法示例:

matrix = [[1, 2, 3], [4, 5], [6, 7, 8, 9]]
result = column_sum(matrix)
print(result)
# 输出: [11, 14, 11, 9]

在上述示例中,我们使用column_sum()函数计算了一个不均匀大小的矩阵(matrix)的每一列的总和。输出的结果表明,第一列的总和是11,第二列的总和是14,第三列的总和是11,第四列的总和是9。

这是一个十分实用的技巧,可以取代手动计算每列的总和。