【问题标题】:Accumulated sum of 2D array [duplicate]二维数组的累计和[重复]
【发布时间】: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


    【解决方案1】:

    np.cumsum 就是你要找的东西:

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

    【讨论】:

      猜你喜欢
      • 2019-05-06
      • 2017-11-15
      • 2016-01-28
      • 2013-04-10
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2017-02-25
      相关资源
      最近更新 更多