[英]pandas groupby weighted cumulative sum
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$$
您希望在groupby
apply
指數加權移動平均值
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
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.