📅  最后修改于: 2023-12-03 15:19:11.482000             🧑  作者: Mango
在Python中,我们可以通过一些简单的代码来获取二维数组中每列的平均值,并将结果返回为一个列表。下面是一个实现该功能的示例代码:
def get_column_averages(matrix):
num_rows = len(matrix)
num_cols = len(matrix[0])
averages = []
for j in range(num_cols):
column_sum = 0
for i in range(num_rows):
column_sum += matrix[i][j]
column_average = column_sum / num_rows
averages.append(column_average)
return averages
以上代码定义了一个名为 get_column_averages
的函数,它接受一个二维数组 matrix
作为参数,并返回一个包含每列平均值的列表 averages
。
首先,我们使用 len()
函数获取二维数组的行数和列数,分别保存到变量 num_rows
和 num_cols
中。
然后,我们初始化一个空列表 averages
,用于保存每列的平均值。
接下来,我们使用两个嵌套的 for
循环遍历二维数组中的每个元素。外层循环按列索引进行迭代,内层循环按行索引进行迭代。
在内层循环中,我们累加每列的元素值,最终得到列的总和。
然后,我们将总和除以行数,得到该列的平均值,并将其添加到 averages
列表中。
最后,在循环结束后,我们将 averages
列表作为函数的返回值。
为了使用这个函数,你可以传递一个二维数组作为参数,例如:
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
averages = get_column_averages(matrix)
print(averages) # 输出 [4.0, 5.0, 6.0]
以上代码将输出一个包含每列平均值的列表 [4.0, 5.0, 6.0]
。
希望以上代码可以帮助你实现获取二维数组中每列平均值的功能!