下面是一个示例代码,可以实现按列求和,并以逗号分隔行ID:
# 定义一个二维列表
matrix = [
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]
# 获取矩阵的行数和列数
rows = len(matrix)
cols = len(matrix[0])
# 按列求和,并以逗号分隔行ID
result = []
for j in range(cols):
col_sum = sum(matrix[i][j] for i in range(rows))
row_ids = ','.join(str(i) for i in range(rows))
result.append(f"{row_ids}:{col_sum}")
# 输出结果
for item in result:
print(item)
以上代码会输出以下结果:
0,1,2:12
0,1,2:15
0,1,2:18
每一行的结果表示该列的和以及参与求和的行ID,行ID通过逗号进行分隔。
上一篇:按列求和限制