numpy的sum函数的axis和keepdim参数详解
一.axis
axis就是指定轴。
三维数组可看作元素是二维数组的一维数组,二维数组可看作元素是一维数组的一维数组。(这么理解就舒服了!)
例:
axis=2就是三维数组对最里面那一层即每个一维数组内部进行求和。
axis=0就是对最外面那一层的元素之间作和。
例子戳这里
二.keepdim
可以理解为'keepdims=True'参数是为了保持结果的维度与原始array相同,即keepdimension保持维度.
importnumpyasnp b=np.arange(12) b=b.reshape(2,6) print(b) print('b中的元素之和:',np.sum(b)) #即在b的第一个轴上进行加和,相当于压缩行,也可以理解为二维矩阵的第一层括号里的东西加和 #若axis=1则是压缩列,也就是对第二层括号里面的进行求和 sum=np.sum(b,axis=0,keepdims=True) print(sum)
运行结果:
[[0 1 2 3 4 5]
[6 7 8 91011]]
b中的元素之和:66
[[6 810121416]]
最后一个输出指定axis=0,keepdim=True,可看到输出的是一个二维数组,如果不加keepdim=True,那么结果就是一维数组[ 6 810121416]
到此这篇关于numpy的sum函数的axis和keepdim参数详解的文章就介绍到这了,更多相关numpy的sum函数axis和keepdim参数内容请搜索毛票票以前的文章或继续浏览下面的相关文章希望大家以后多多支持毛票票!
声明:本文内容来源于网络,版权归原作者所有,内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:czq8825#qq.com(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。