熊猫移动平均

Question

我有数据：

date  count 
2015-09-01 5 
2015-09-02 4 
2015-09-03 8 
2015-09-04 8 
2015-09-05 3 
2015-09-06 5 
2015-09-07 9 
2015-09-08 7 
2015-09-09 5 
2015-09-10 7 
... 

我需要得到均线在过去的5天。

我该怎么做python和pandas？

Answer 1

IIUC你想rolling_mean：

In [136]: 

df.set_index('date', inplace=True) 
pd.rolling_mean(df['count'], window=5) 
Out[136]: 
date 
2015-09-01 NaN 
2015-09-02 NaN 
2015-09-03 NaN 
2015-09-04 NaN 
2015-09-05 5.6 
2015-09-06 5.6 
2015-09-07 6.6 
2015-09-08 6.4 
2015-09-09 5.8 
2015-09-10 6.6 
dtype: float64 

Answer 2

大熊猫v0.22.0，你可以使用rolling

df.rolling(5).mean()