小编ash*_*huk的帖子

如何在Python Polars中的groupby之后分配指数移动平均线

我刚刚开始在 python 中使用极坐标,我来自 pandas。我想知道如何在 python Polars 中复制下面的 pandas 代码

import pandas as pd
import polars as pl

df['exp_mov_avg_col'] = df.groupby('agg_col')['ewm_col'].transform(lambda x : x.ewm(span=14).mean())
Run Code Online (Sandbox Code Playgroud)

我已经尝试过以下方法:

df.groupby('agg_col').agg([pl.col('ewm_col').ewm_mean().alias('exp_mov_avg_col')])
Run Code Online (Sandbox Code Playgroud)

但这给了我每个提供者的指数移动平均值列表,我希望将该列表分配给原始数据帧中正确索引的列,就像 pandas 代码一样。

python-polars

3
推荐指数
1
解决办法
1265
查看次数

标签 统计

python-polars ×1