如何解决二维数组的累计和
假设我有一个像下面这样的 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)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。