关于numpy mean函数的axis参数

来源:互联网 发布:mac word转pdf 超链接 编辑:程序博客网 时间:2024/06/05 22:35

http://www.jianshu.com/p/f9e3fd264932

理解多维矩阵的"求和"、"平均"操作确实太恶心了,numpy提供的函数里还有一堆参数,搞得晕头转向的,这里做个笔记,提醒一下自己, 下面是例程


import numpy as npX = np.array([[1, 2], [4, 5], [7, 8]])print np.mean(X, axis=0, keepdims=True)print np.mean(X, axis=1, keepdims=True)

结果是分别是

                 [[ 1.5] [[ 4.  5.]]      [ 4.5]                      [ 7.5]]

我个人比较raw的认识就是,axis=0,那么输出矩阵是1行,求每一列的平均(按照每一行去求平均);axis=1,输出矩阵是1列,求每一行的平均(按照每一列去求平均)。还可以这么理解,axis是几,那就表明哪一维度被压缩成1。