计算数组中变量的平均值，以获取另一个变量的设定值

Question

我希望计算一系列深度的平均温度（所以平均温度为2米，然后是3米等），这样我最终能够创建平均温度与深度的关系图。温度是数组中的一列，深度是另一列。由于存在一系列深度（从2米到> 200米），我不想手动输入每个深度的代码，所以我想知道如何做到这一点。这就是我到目前为止所做的：

import numpy as np
temperature = data['Temperature']
depth = data['Depth']

meantemp = np.mean(temperature)

Answer 1

下载/复制并将找到的代码here保存为（f.i.）grouping.py：

然后写：

from grouping import group_by
print group_by(depth.round(0)).mean(temperature)