【发布时间】:2021-01-15 17:36:46
【问题描述】:
假设我有一个像下面这样的二维 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])
【问题讨论】:
标签: python python-3.x numpy numpy-ndarray numpy-einsum