为几列编写 pandas 移动平均值的更简洁的方法

Question

I have a df with four columns as below:

A	B	C	D
2	55	45	23
3	78	78	20

...

I would like to take a moving average of all of the columns with new names, and keep the original columns too. I can do this using:

cols=['A','B','C','D']

df[cols].rolling(window=28, min_periods=14).mean().join(df, rsuffix='x').rename(columns={'A':'28d_A','B':'28d_B','C':'28d_C','D':'28d_D'})

Is there a better way which doesn't require me to rename all the columns and do a join? Thanks

Answer 1

You can us a dict comprehension to assign() . Sample data is too short for output to be meaningful.

cols=['A','B','C','D']
df.assign(**{f"28d_{c}":df[c].rolling(window=28, min_periods=14).mean() for c in cols})
``