python numpy中cumsum的用法详解
Cumsum：计算轴向元素累加和，返回由中间结果组成的数组
重点就是返回值是“由中间结果组成的数组”
以下代码在python3.6版本运行成功！
下面看代码，定义一个2*2*3的数组，所以其shape是2，2，3，索引分别0，1，2
shape
索引
2
0
2
1
3
2
代码：
importnumpyasnp arr=np.array([[[1,2,3],[8,9,12]],[[1,2,4],[2,4,5]]])#2*2*3 print(arr.cumsum(0)) print(arr.cumsum(1)) print(arr.cumsum(2))
输出结果:
#cumsum(0)
[[[1 2 3]
[8 912]]
[[2 4 7]
[101317]]]
#cumsum(1)
[[[1 2 3]
[91115]]
[[1 2 4]
[3 6 9]]]
#cumsum(2)
[[[1 3 6]
[81729]]
[[1 3 7]
[2 611]]]
注释：
- arr是一个2*2*3三维矩阵，索引值为0，1，2
- cumsum(0)：实现0轴上的累加：以最外面的数组元素为单位，以[[1,2,3],[8,9,12]]为开始实现后面元素的对应累加
- cumsum(1)：实现1轴上的累加：以中间数组元素为单位，以[1,2,3]为开始，实现后面元素的对应累加
- cumsum(2)：实现2轴上的累加：以最里面的元素为累加单位，即1为开始，实现后面的元素累加
四维数组实现
下面看一个四维数组2*2*2*4，索引值为0，1，2，3
代码：
importnumpyasnp arr=np.arange(32).reshape((2,2,2,4)) print(arr) print(arr.cumsum(0)) print(arr.cumsum(1)) print(arr.cumsum(2)) print(arr.cumsum(3))
arr: [[[[0123] [4567]] [[891011] [12131415]]] [[[16171819] [20212223]] [[24252627] [28293031]]]]
arr是一个2*2*2*4四维矩阵，索引值为0，1，2，3
cumsum(0)：实现0轴上的累加即：以最外面数组元素为单位即
[[[0123] [4567]] [[891011] [12131415]]]
与
[[[16171819] [20212223]] [[24252627] [28293031]]]]
对应位置元素相加起来
结果：
[[[[0123] [4567]] [[891011] [12131415]]] [[[16182022] [24262830]] [[32343638] [40424446]]]]
cumsum(1)：实现1轴上的累加即：以次外面元素为单位，累加：
[[0123] [4567]]
加
[[891011] [12131415]]
加
[[16171819] [20212223]]
加
[[24252627] [28293031]]
累计过程产生的中间结果要记录到数组中
所以，结果：
[[[[0123] [4567]] [[8101214] [16182022]]] [[[16171819] [20212223]] [[40424446] [48505254]]]]
cumsum(2)就对应从[0 1 2 3]数组元素开始实现累加，然后记录中间结果
cumsum(3)对应的是从最里面最小的数组元素，即从0开始实现累加，记录中间结果
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。