大Pandas倒向时间窗均值的计算

Question

不确定我的题目是否正确，请建议编辑

我有以下数据帧：

df_dict={'startDate': 
['2015-01-01','2015-01-05','2015-01-18','2015-01-25',
'2015-02-02','2015-02-03','2015-02-04','2015-02-17',
'2015-03-03','2015-03-04','2015-03-07','2015-03-11',
'2015-04-04','2015-04-05','2015-04-17','2015-04-19'],
'price':[200,250,270,240,100,120,130,140,300,310,330,370,400,410,420,440],

'250_index': [1,2,1,2,1,2,1,2,1,2,1,2,1,2,1,2]}
df=pd.DataFrame(df_dict)

我需要在我的数据集中找到第一个可用的日期，在本例中是2015-01-01，然后创建30day区间，例如：2015-01-01，2015-01-31，2015-03-02，2015-04-01，然后对于每个区间和250_index组合计算过去的30day60_day和90day平均价格。你知道吗

结果数据帧应该是这样的，第一个区间的过去平均值是0，因为我们没有关于发生了什么的信息，但是对于第二个区间2015-01-31，我们需要根据2015-01-01和2015-01-31之间的价格来计算past_30_avg，对于其他区间也是如此：

time_interval    index    past_30_avg  past_60_avg  past_90_avg
 '2015-01-01'     1         0             0            0
                  2         0             0            0
 '2015-01-31'     1         235           0            0
                  2         245           0            0
 '2015-03-02'     1         115          175           0
                  2         130          187           0
 '2015-04-01'     1         315          240          221    
                  2         340          235          238

我正在寻找一个通用的方法，因为它只是我数据集的一部分

Answer 1

考虑^{}表示30天的平均值（每月汇总），然后^{}表示60天和90天^{}：

tdf = df.groupby([pd.Grouper(key='startDate', freq='30D', label='right'), '250_index']).mean()\
              .reset_index().rename(columns={'price': 'past_30_avg'})
print(tdf)

tdf['past_60_avg'] = tdf.groupby('250_index')['past_30_avg'].transform(lambda x: x.rolling(2).mean())
tdf['past_90_avg'] = tdf.groupby('250_index')['past_30_avg'].transform(lambda x: x.rolling(3).mean())

#    startDate  250_index  past_30_avg  past_60_avg  past_90_avg
# 0 2015-01-31          1          235          NaN          NaN
# 1 2015-01-31          2          245          NaN          NaN
# 2 2015-03-02          1          115        175.0          NaN
# 3 2015-03-02          2          130        187.5          NaN
# 4 2015-04-01          1          315        215.0   221.666667
# 5 2015-04-01          2          340        235.0   238.333333
# 6 2015-05-01          1          410        362.5   280.000000
# 7 2015-05-01          2          425        382.5   298.333333