熊猫分组加权累计和

Question

A   | B    | B_new
----| ---- |---------
1   | 1    |  1
1   | 2    |  1.5
1   | 3    |  2.25
2   | 2    |  2
2   | 4    |  3
2   | 6    |  4.5

我有一个数据框，我想按A计算B组的加权累计和。但是我不知道如何应用变换。

$$new = C*cur+(1-C)*old$$

Answer 1

您想要在一段时间内进行apply 指数加权移动平均groupby

df.assign(
    B_new=df.groupby('A').B.apply(
        lambda x: x.ewm(alpha=0.5, adjust=False).mean()
    )
)

   A  B  B_new
0  1  1   1.00
1  1  2   1.50
2  1  3   2.25
3  2  2   2.00
4  2  4   3.00
5  2  6   4.50