python中cumsum函数_python cumsum函数

这个函数的功能是返回给定axis上的累计和

函数的原型如下：详见doc

numpy.cumsum(a, axis=None, dtype=None, out=None)

Return the cumulative sum of the elements along a given axis.

官方的文档是没有详细解释，本人做了测试自己把理解的写下。

1.对于一维输入a(可以是list，可以是array，假设a=[1, 2, 3, 4, 5, 6, 7] ，就是当前列之前的和加到当前列上，如下：

>>>import numpy as np

>>> b=[1,2,3,4,5,6,7]

>>> np.cumsum(a)

array([ 1, 3, 6, 10, 15, 21, 28, 36, 45, 55, 75, 105])

2.对于二维输入a，axis可以是0(第一行不动，其他行累加)， 1(第一列列不动，其他列累加)，如下：

>>>import numpy as np

>>> c=[[1,2,3],[4,5,6],[7,8,9]]

>>> np.cumsum(c,axis=0)

array([[ 1, 2, 3],

[ 5, 7, 9],

[12, 15, 18]])

>>> np.cumsum(c,axis=1)

array([[ 1, 3, 6],

[ 4, 9, 15],

[ 7, 15, 24]])

3.对于三维输入a, axis可以是 0(第0维不动，其他维累加)， 1(第1维不动，其他维累加)， 2(第2维不动，其他维累加)，注意维数从外到内是0-2编号，如下：

>>>import numpy as np

>>> a

[[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 20, 30]]]

>>> np.cumsum(a,axis=0)

array([[[ 1, 2, 3],

[ 4, 5, 6]],

[[ 8, 10, 12],

[14, 25, 36]]])

>>> np.cumsum(a,axis=1)

array([[[ 1, 2, 3],

[ 5, 7, 9]],

[[ 7, 8, 9],

[17, 28, 39]]])

>>> np.cumsum(a,axis=2)

array([[[ 1, 3, 6],

[ 4, 9, 15]],

[[ 7, 15, 24],

[10, 30, 60]]])

a是一个 2 x 2 x 3 的 list，解释起来有点麻烦，先把 a 变种风格

a=[ //axis=0

[//axis=1

[1, 2, 3],//axis=2

[4, 5, 6] //axis=2

],

[//axis=1

[7, 8, 9],//axis=2

[10, 20, 30] //axis=2

],

]

axis=0说明，以对外层的第0行(也就是由1,2,3,4,5,6组成的list)不动，累加到其他行，注意是累加，累加，累加。重要事说三遍

axis=1说明，第2维度的第0行不变，以就是 [1,2,3]和[7,8,9]不动，累加到其他行

axis=2说明，第3维度，也是最内层，此时是第0列保持不变，见蓝色数字，累加到其他列上，注意只有最内层维度才是列不变

对于更高维度，可以参考3维来从外向内剥离的方式理解

参考文献

cumsum官方doc