📅  最后修改于: 2023-12-03 15:19:17.373000             🧑  作者: Mango
当处理不均匀大小的列表时,可能需要计算每一列的总和。在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。
这是一个十分实用的技巧,可以取代手动计算每列的总和。