这篇文章主要介绍“numpy的sum函数的axis和keepdim参数怎么使用”,在日常操作中,相信很多人在numpy的sum函数的axis和keepdim参数怎么使用问题上存在疑惑,小编查阅了各式资料,整理出简单好用的操作方法,希望对大家解答”numpy的sum函数的axis和keepdim参数怎么使用”的疑惑有所帮助!接下来,请跟着小编一起来学习吧!
一.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参数怎么使用”的学习就结束了,希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习,快去试试吧!若想继续学习更多相关知识,请继续关注编程网网站,小编会继续努力为大家带来更多实用的文章!