以下是一个按行分组并求和的代码示例:
data = [
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]
result = []
for row in data:
row_sum = sum(row)
result.append(row_sum)
print(result)
输出:
[6, 15, 24]
在上述代码中,我们首先定义了一个二维列表 data
,其中包含了三行三列的数据。然后,我们创建一个空列表 result
,用于存储每行求和的结果。
接下来,我们使用 for
循环遍历 data
中的每一行。对于每一行,我们使用 sum()
函数求出该行元素的和,并将结果存储到变量 row_sum
中。然后,我们将 row_sum
添加到 result
列表中。
最后,我们打印输出 result
列表,即为每行求和的结果。
上一篇:按行分区的稀疏矩阵-向量乘法
下一篇:按行分组但给出其他行的最高值