二维数组的累计和

问题描述

假设我有一个像下面这样的 2D numpy 数组

dat = np.array([[1,2],[3,4],[5,6],[7,8])

我想得到一个新数组，每行等于其前一行与自身的总和，如下所示

first row: [1,2]
second row: [1,2] + [3,4] = [4,6]
third row: [4,6] + [5,6] = [9,12]
forth row: [9,12] + [7,8] = [16,20]

所以数组就像

dat = np.array([[1,[4,[9,12],[16,20])    

解决方法

np.cumsum 就是您要找的：

dat = np.array([[1,2],[3,4],[5,6],[7,8]])
result = np.cumsum(dat,axis=0)