二维数组的累加和
Accumulated sum of 2D array
假设我有一个如下所示的 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,2],[4,6],[9,12],[16,20])
np.cumsum 就是你要找的:
dat = np.array([[1,2],[3,4],[5,6],[7,8]])
result = np.cumsum(dat, axis=0)
假设我有一个如下所示的 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,2],[4,6],[9,12],[16,20])
np.cumsum 就是你要找的:
dat = np.array([[1,2],[3,4],[5,6],[7,8]])
result = np.cumsum(dat, axis=0)