下面是一个编写计算行均值的函数的示例代码:
def calculate_row_average(matrix):
row_averages = []
for row in matrix:
row_sum = sum(row)
row_average = row_sum / len(row)
row_averages.append(row_average)
return row_averages
# 测试代码
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
averages = calculate_row_average(matrix)
print(averages) # 输出 [2.0, 5.0, 8.0]
该函数接受一个二维矩阵作为参数,遍历矩阵的每一行,计算每一行元素的和并除以行长度来得到行平均值,然后将行平均值添加到一个新的列表中。最后,返回行平均值列表。
在上述示例中,我们定义了一个3x3的矩阵,并将其作为参数传递给calculate_row_average
函数。函数计算出每一行的平均值,并将结果打印出来。输出结果是[2.0, 5.0, 8.0]
,表示第一行的平均值为2.0,第二行的平均值为5.0,第三行的平均值为8.0。