Pandas按组加权累积和

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$$

- Lehyu

Answer 1

你想在 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

- piRSquared