Python：Numpy中的cumsum()函数的使用

跳舞的皮埃尔

已于 2023-07-12 10:33:36 修改

阅读量3.6k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Python 笔记 numpy 文章标签： numpy python 开发语言

于 2023-04-07 10:43:10 首次发布

101 篇文章

订阅专栏

28 篇文章

订阅专栏

1 篇文章

订阅专栏

cumsum函数用于计算数组的累计求和，当不指定axis时，对整个数组进行求和；axis=0表示按行累加，axis=1表示按列累加。示例展示了在不同axis设置下的运算结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、作用

cumsum 的作用主要就是计算轴向的累加和。

def cumsum(self, axis=None, dtype=None, out=None):
	"""
	:param axis:
	:param dtype:
	:param out:
	"""

假设二维数组：

在这里插入图片描述

不指定 axis 参数时，把二维数组当作了一维数组处理，进行累计求和运算。即1,1+2,1+2+3,1+2+3+4,1+2+3+4+5,1+2+3+4+5+6

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6]])
np.cumsum(a)

输出：

array([ 1,  3,  6, 10, 15, 21], dtype=int32)

在这里插入图片描述

axis=0 指的是按行累加，即 本行 = 本行 + 上一行，即：5=1+4,7=2+5,9=3+6

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6]])
np.cumsum(a, axis=0)

输出：

array([[1, 2, 3],
       [5, 7, 9]], dtype=int32)

在这里插入图片描述

可以看出，上述代码是按照轴0进行累计求和的。

参数 axis=1 指的是按列相加，即 本列 = 本列 + 上一列，即：3=1+2，6=1+2+3，9=4+5，15=4+5+6

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6]])
np.cumsum(a, axis=1)

输出：

array([[ 1,  3,  6],
       [ 4,  9, 15]], dtype=int32)

在这里插入图片描述
可以看出，上述代码是按照轴1进行累计求和的。

关于更高维的数组的运算我就暂时就不测试了，暂时也用不上。大伙可以在评论区探讨一下。