当前位置:   article > 正文

理解numpy mean(axis=?)_x.mean(axis=0)

x.mean(axis=0)

就是为了记载下来防止忘记,没什么技术含量。其实就是求平均值。

参考博客:http://www.cnblogs.com/ymjyqsx/p/6495450.html

  1. >>X = np.float32(12,32,34,54)
  2. >>print X
  3. [12.32.34.54]
  4. >>print X.mean(axis=0)
  5. 33.0
  6. >>print X.mean(axis=1)
  7. Index Error: tuple index out of range

再来一个小例子

  1. >>X=np.array([[1,2],[3,4],[5,6]])
  2. >>print X
  3. [[1 2]
  4. [3 4]
  5. [5 6]]
  6. >>print X.mean(axis=0)
  7. [3. 4.]
  8. >>print X.mean(axis=1)
  9. [1.5 3.5 5.5]
  10. >>print np.mean(X,axis=0,keepdims=True)
  11. [[3. 4.]]
  12. >>print np.mean(X,axis=0,keepdims=False)
  13. [3. 4.]
  14. >>print np.mean(X,axis=1,keepdims=True)
  15. [[1.5]
  16. [3.5]
  17. [5.5]]
  18. >>print np.mean(X,axis=1,keepdims=False)
  19. [1.5 3.5 5.5]

对于这个例子需要解释:

axis=0表示输出矩阵是1行,也就是求每一列的平均值。(1+3+5)/3=3,(2+4+6)/3=4

axis=1表示输出矩阵是1列, 也就是求每一行的平均值,(1+2)/2=1.5,(3+4)/2=3.5,(5+6)/2=5.5

实际上这个axis=0就是选择shape中第一个元素(即第一维)变为1,axis=1就是选择shape中第二个元素变为1。用shape来看会比较方便。

举个例子:

  1. >>> x = np.array([[1, 2], [4, 5], [7, 8]])
  2. >>> x.shape
  3. (3, 2)
  4. >>> y = np.mean(x, axis=0, keepdims=True)
  5. >>> y.shape
  6. (1, 2)

比如我们输入为batch = [128, 28, 28],可以理解为batch=128,图片大小为28×28像素,我们相求这128个图片的均值,应该这么写

m = np.mean(batch, axis=0)

也就相当于shape的第0个元素被压缩成了一维度,那么在这里意思就是把128张图片压成了一个,求得是它们在每一个像素点的平均值。

如果不给出axis会是什么情况呢?其实求所有值的均值,例子:

  1. >>print X
  2. [[1 2]
  3. [3 4]
  4. [5 6]
  5. ]
  6. >>print X.mean()
  7. 3.5

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/花生_TL007/article/detail/133135
推荐阅读
相关标签
  

闽ICP备14008679号