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

一、axis

axis参数用于指定进行求和操作的轴。

理解数组结构时，可将三维数组视为由二维数组组成的一维数组，将二维数组视为由一维数组组成的一维数组。这种层级视角有助于直观理解求和过程。

具体示例如下：

当设置axis=2时，表示对三维数组中最内层的一维数组内部元素进行求和。

当设置axis=0时，表示对最外层数组中的各个元素（即子数组）之间进行求和。

相关示例代码可参考此处。

二、keepdim

keepdims参数设为True时，旨在保持输出结果的维度数与输入数组的维度数相同，即保持维度（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参数内容请搜索本站以前的文章或继续浏览下面的相关文章希望大家以后多多支持本站！