以下是一个示例代码,演示如何按列计算一个二维数组的总和。
def column_sums(matrix):
# 获取矩阵的行数和列数
rows = len(matrix)
cols = len(matrix[0])
# 创建一个列表来存储每列的总和
sums = [0] * cols
# 遍历每一列
for j in range(cols):
# 遍历每一行,并将当前列的值累加到总和中
for i in range(rows):
sums[j] += matrix[i][j]
return sums
# 测试
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
result = column_sums(matrix)
print(result) # 输出 [12, 15, 18]
在上述代码中,column_sums
函数接收一个二维数组作为输入参数。首先,它获取矩阵的行数和列数,并创建一个与列数相同长度的列表来存储每列的总和。然后,它使用两个嵌套的循环来遍历每个元素,并将它们累加到相应列的总和中。最后,返回计算得到的总和列表。
在给定的示例中,矩阵 [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
的每列总和分别为 [12, 15, 18]
。
上一篇:按列计算子总计数的度量
下一篇:按列基于出现次数分组