python中按id/组按定义间隔移动平均

Question

给定的是一个csv数据集，包含以下列和值（示例）。 我想在csv数据中添加一个新的列，根据每个id，移动平均值为S1。S1是我的度量值。时间范围t应该是3。这就是我的数据集当前的样子。你知道吗

id  S1  
1   3
1   4
1   2
1   6
1   9
2   3
2   1
2   2
2   3
2   8
2   6
3   1
3   4
3   2
3   8
3   5

这就是我想做的：

id  S1   movA  
1   3    NaN
1   4    NaN
1   2    3.000
1   6    4.000
1   9    5.667
2   3    NaN
2   1    NaN
2   2    2.000
2   3    2.000
2   8    4.333
2   6    5.333
3   1    NaN
3   4    NaN
3   2    2.333
3   8    4.667
3   5    5.000

Answer 1

与mean一起使用groupby和rolling

g = df.groupby('id').S1
rolls = [2, 3, 4, 5]
pd.concat([g.rolling(i).mean() for i in rolls], axis=1, keys=rolls)