一列中几个选定行的总和

时间:2019-01-13 19:01:55

标签: python pandas column-sum

请考虑以下数据框:

     list1  

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

我想用熊猫编写代码,其中“ 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  

1 个答案:

答案 0 :(得分:0)

rollingsum一起使用:

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