numpy mean(data, axis) 中 axis 的设置

还是感觉怪怪的,欢迎大家指教

numpy mean(data, axis)

其中axis=0为按列求平均,axis=1为按行求平均,不给出axis不是默认axis为0,而是把所有元素加起来求平均.

原始矩阵的shape=[3,4,5],取axis=0再进行操作后,得到的矩阵shape=[4,5]。同样的,取axis=1再进行操作后,得到的矩阵shape=[3,5]。取axis=-1(axis=2)再操作后,shape=[3,4]

1表示横轴,方向从左到右;0表示纵轴,方向从上到下。当axis=1时,数组的变化是横向的,体现出列的增加或者减少。反之,当axis=0时,数组的变化是纵向的,体现出行的增加或减少。

矩阵

>>X=np.array([[1,2],[3,4],[5,6]])
>>print X
>>[[1 2]
 [3 4]
 [5 6]]

axis = 0
X=
[[1 2]
[3 4]
[5 6]]
向下排列矩阵中元素,每列各自求mean

(1+3+5)/3=3,(2+4+6)/3=4

>>print X.mean(axis=0)
[3. 4.]

axis = 1
X = [[1 2] [3 4] [5 6]]
向右排列矩阵中元素,每行各自求mean

(1+2)/2=1.5,(3+4)/2=3.5,(5+6)/2=5.5

>>print X.mean(axis=1)
[1.5 3.5 5.5]

不给出 axis
求所有值的均值

>>print X
[[1 2]
 [3 4]
 [5 6]
]
>>print X.mean()
3.5
上一篇:求平均值(C语言)


下一篇:R语言非参数方法:使用核方法和K-NN(k近邻算法)分类预测心脏病数据