计算滚动（平均差）^2 熊猫

Question

您好，我正在尝试使用滚动 3 天平均值在数据框中创建一个“均差平方”列。

结果明智我想从上到下（NaN、NaN、26、8、8、8）获得具有以下值的第 3 列，其中 26 例如计算为 (1-5)^2 + (6- 5)^2 + (8-5)^2，均方值8开始(6-8)^2....等等。

我已经编写了前两行代码，不知道如何编写进一步的代码来达到预期的效果。任何帮助表示赞赏。

df = pd.DataFrame({'Data':[1, 6, 8, 10, 12, 14]})
df['mean'] = df.rolling(window=3).mean()
df['mean difference_squared'] = ........

Answer 1

你的情况

df.Data.rolling(window=3).apply(lambda x : sum((x-x.mean())**2),raw=True)
Out[173]: 
0     NaN
1     NaN
2    26.0
3     8.0
4     8.0
5     8.0
Name: Data, dtype: float64