numpy的sum函数的axis和keepdim参数详解

时间:2021-05-23

一.axis

axis就是指定轴。

三维数组可看作元素是二维数组的一维数组,二维数组可看作元素是一维数组的一维数组 。(这么理解就舒服了!)

例:

axis=2就是三维数组对最里面那一层即每个一维数组内部进行求和。

axis=0就是对最外面那一层的元素之间作和。

例子戳这里

二.keepdim

可以理解为'keepdims = True'参数是为了保持结果的维度与原始array相同,即keep dimension 保持维度.

import numpy as np 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 9 10 11]]
b中的元素之和: 66
[[ 6 8 10 12 14 16]]
最后一个输出指定axis=0,keepdim=True,可看到输出的是一个二维数组,如果不加keepdim=True,那么结果就是一维数组[6 8 10 12 14 16]

到此这篇关于numpy的sum函数的axis和keepdim参数详解的文章就介绍到这了,更多相关numpy的sum函数axis和keepdim参数内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章