以下是一个按行加和并进行标记的示例代码:
def sum_and_mark_rows(matrix):
# 初始化一个用于存放结果的列表
result = []
# 遍历矩阵的每一行
for row in matrix:
# 计算当前行的和
row_sum = sum(row)
# 将行和添加到结果列表中
result.append(row_sum)
# 标记当前行(这里以行和大于10为例)
if row_sum > 10:
row.append("标记")
# 返回结果列表
return result
# 测试代码
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
result = sum_and_mark_rows(matrix)
print(result) # 输出: [6, 15, 24]
print(matrix) # 输出: [[1, 2, 3, '标记'], [4, 5, 6], [7, 8, 9, '标记']]
在上述示例代码中,sum_and_mark_rows
函数接受一个二维矩阵作为参数,并返回一个包含每行和的列表。在计算每行和时,我们使用了 sum
函数来对当前行进行求和。然后,我们将行和添加到结果列表中,并根据特定的条件进行标记(这里以行和大于10为例)。最后,我们返回结果列表。
在测试代码中,我们创建了一个示例矩阵 matrix
,并使用 sum_and_mark_rows
函数对其进行处理。结果列表 result
包含了每行的和,而矩阵 matrix
在满足条件的行末尾添加了标记。打印结果如注释所示。
下一篇:按行将矩阵乘以向量