python - 在 Python 中计算滚动总和

Question

我有一个具有以下格式的数据框 df:

    Type     2013         2014          2015         2016          2017         
           EU  NO_EU    EU   NO_EU    EU  NO_EU    EU  NO_EU     EU  NO_EU               
     A      4     6      2     1       2    7       1    4        2    6    

     B      3     0      8     3       11   22      2    2        2    1   

我想创建一个新的数据框，其中包含 EU 和 NO_EU 3 年期间的滚动总和。因此，我的新数据框应该是这样的:

           Type     2013         2014          2015         2016          2017         
                  EU  NO_EU    EU   NO_EU    EU  NO_EU    EU  NO_EU     EU  NO_EU     

            A      -     -      -      -      8    14      5    12       5    17     
            B      -     -      -      -      22   25      21   27       15   25      

前两年将没有数据，因为我们没有足够的数据来计算过去 3 年的总和。

关于如何在 python 中做到这一点有什么想法吗？也许使用类似于 df.rolling(window=3).sum 的东西？我的实际数据框有很多行和列，因此我将使用循环来计算它们。

感谢您提前提供的帮助

Answer 1

您可以尝试使用rolling跨axis=1，即沿列，而不是沿行axis=0。

df.rolling(window=3,axis=1).sum()