Pandas：添加“对角线”

Question

我有一个数据帧，其中索引是time，但每行的数据不是同质的w.r.t.time，如下所示：

                   obs_TQ       obs_PQ    
year quarter                          
1    1        5499.330651  1655.414960
     2        4650.849431  1807.889632
     3        5417.065129  3352.855656
     4        7771.021167  2343.369804

其中obs_TQ是本季度的观察值，obs_PQ是上一季度的观察值。我想按季度总结观察结果，因此按季度得出的结果是

                                   real_obs
year quarter
   1       1      obs_TQ[1,1] + obs_PQ[1,2]
   1       2      obs_TQ[1,2] + obs_PQ[1,3]

也就是说，对于每一行，我需要将该行的obs_TQ添加到下一行的obs_PQ：

                     real_obs
year quarter
   1       1      7307.220283
   1       2      8003.705087

有没有有效的矢量化方法？按索引分组在这里不起作用，因为我需要组合来自不同组的数据，而且我不知道在这种情况下还可以使用任何其他工具

Answer 1

除非我漏掉了什么，你就不能把TQ加到移位的PQ上吗

>>> (df["obs_TQ"] + df["obs_PQ"].shift(-1)).to_frame("real_obs")
                 real_obs
year quarter             
1    1        7307.220283
     2        8003.705087
     3        7760.434933
     4                NaN