python numpy中axis和keepdims的含义

最近本蓝在学习的时候发现numpy中有一些参数的设置比较有意思,于是我就想整明白这些设置到底是啥。比如axis和keepdims。
axis有两个参数,分别是axis=0,axis=1.本蓝一开始认为0代表行,1代表列,可惜啊,想反了。axis其实是延数轴方向的延申,啥意思呢,上图:

在这里插入图片描述
这样是不是就迎刃而解了,哈哈哈哈哈。
再看keepdims,字面上看就是保持维度,实际上也是这个作用,一般用法为keepdims = True

import numpy as np
a = np.array([[1,10],[2,8]])
print(np.sum(a,axis=1,keepdims=True))#add by line

输出结果为:
[[11]
[10]]

猜你喜欢

转载自blog.csdn.net/weixin_44843629/article/details/106742245