Pandas:使用3个周期的窗口进行计算

Question

我想知道如何计算下一个关系。公式是这样做的:它取最后三行，然后用该数据计算公式。并使用滚动窗口3来完成此操作

A   B   Formula Result
2   3     nan
3   4     nan
2   3    sqrt(3*0.33)
1   2    sqrt(3*0.33)
3   1    sqrt(6*0,33)
1   2    sqrt(6*0,33)

公式：

​

​

​

我尝试过的：

df["Formula Result"] = sum((df["A"]-Df["B"])^2).rolling_window(3)

Answer 1

df["Formula Result"] = (df.A
                          .rolling(3)
                          .apply(lambda x: np.sqrt(0.33) * np.linalg.norm(x - df.loc[x.index, "B"])))

在A上滚动，然后在滚动窗口的索引上到达相应的B值。该公式对应于两个向量之间的欧几里德距离，因此我们可以使用差的范数。在方括号内乘以0.33相当于从外部将其与sqrt(0.33)相乘。

也可以在没有np.linalg.norm的情况下编写它

df["Formula Result"] = (df.A
                          .rolling(3)
                          .apply(lambda x: np.sqrt(sum(0.33 * (x - df.loc[x.index, "B"])**2))))

要获得

>>> df

   A  B  Formula Result
0  2  3             NaN
1  3  4             NaN
2  2  3        0.994987
3  1  2        0.994987
4  3  1        1.407125
5  1  2        1.407125