python pandas 计算平均值

Question

我有一个这样的数据框:

        pk_dcdata     threshold   last_ep  diff
window                                                            
1        11075761       0.00001         4     3
1        11075768       0.00001         7     6
2        11075769       0.00001         1    -1
2        11075770       0.00001         1    -1
3        11075771       0.00001         1     0
3        11075768       0.00001         7     6

我想计算“diff”列中的平均值，但与索引“window”进行比较，并将平均值保存到新列表中。例如window = 1 和均值是 (3+6)/2，接下来是 window = 2，所以 (-1-1)/2 等等。

预期结果:list = [4.5,-1,3]

我尝试使用“rolling_mean”但不知道如何设置移动长度。由于数据集很大，希望能有一种快速的方法得到结果。

Answer 1

不要使用 list 作为变量，因为 python 保留字。

需要按每个索引的mean 聚合，最后将Series 转换为list:

L = df.groupby(level=0)['diff'].mean().tolist()
#alternative
#L = df.groupby('window')['diff'].mean().tolist()
print (L)
[4.5, -1.0, 3.0]

替代工作在 pandas 0.20.0+，检查 docs .