列中少数选定行的总和

Question

考虑DATAFAFRAM:如下：

     list1  

 0    2         
 1    5         
 2    4     
 3    8         
 4    4         
 5    7         
 6    8 

我想用pandas编写一个代码，其中“sum”是“list1”中两行后续元素的总和，输出如下：

     list1      sum

 0     2        NaN 
 1     5        7       
 2     4        9       
 3     8        12      
 4     4        12  
 5     7        11      
 6     8        15  

Answer 1

将^{}与sum一起使用：

df['sum'] = df['list1'].rolling(2).sum()
print (df)
   list1   sum
0      2   NaN
1      5   7.0
2      4   9.0
3      8  12.0
4      4  12.0
5      7  11.0
6      8  15.0