python cumsum函数的具体使用
这个函数的功能是返回给定axis上的累计和函数的原型如下:详见doc
numpy.cumsum(a,axis=None,dtype=None,out=None)
Returnthecumulativesumoftheelementsalongagivenaxis.
官方的文档是没有详细解释,本人做了测试自己把理解的写下。
1.对于一维输入a(可以是list,可以是array,假设a=[1,2,3,4,5,6,7],就是当前列之前的和加到当前列上,如下:
>>>importnumpyasnp >>>a=[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行不动,将第1行累加到其他行);axis=1(进入最内层,转化成列处理。第1列不动,将第1列累加到其他列),如下:
>>>importnumpyasnp >>>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(第1行不动,将第1行累加到其他行);axis=1(进入第2层,每个第二层第1行不动,累加到其他行);axis=2(进入最内层,转化成列处理。第1列不动,将其累加到其他列),注意维数从外到内是0-2编号,如下:
>>>importnumpyasnp >>>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是一个2x2x3的list,解释起来有点麻烦,先把a变种风格
[//axis=0 [//axis=1。同时是axis=0所对应的第1行,注意是此中括号包含的整个内容 [1,2,3],//axis=2。同时是axis=1所对的第1行 [4,5,6]//axis=2 ], [//axis=1。同时是axis=0所对应的第2行 [7,8,9],//axis=2。同时是axis=1所对的第1行 [10,20,30]//axis=2 ], ]
axis=0说明,第1维度,包含2行,绿色中括号的list,第1行(也就是由1,2,3,4,5,6组成的list)不动,累加到第二行,注意是累加
axis=1说明,第2维度, 每个绿色中括号里第1行不变,也就是[1,2,3]和[7,8,9]不动,累加到同兄弟行(同属一个中括号的行)
axis=2说明,第3维度,也是最内层,转化成列处理,紫色数字所在列不动,累加到其他列上
对于更高维度,可以参考3维来从外向内剥离的方式理解
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。
声明:本文内容来源于网络,版权归原作者所有,内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:czq8825#qq.com(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。