numpy之函数mean()

mean函数中参数 axis表示在该轴上所有元素求平均的输出

>>> a = np.array([np.arange(4),np.arange(4)])
>>> a
array([[0, 1, 2, 3],
       [0, 1, 2, 3]])
>>> np.mean(a)
1.5
>>> np.mean(a,axis=0)
array([0., 1., 2., 3.])
>>> np.mean(a,axis=1)
array([1.5, 1.5])
>>> 

mean()输入参数之keepdims,

keepdims = True 表示保持原有输入矩阵的维度
keepdims =False 该参数默认为False

>>> a=np.arange(6)
>>> a=a.reshape(3,2)
>>> a
array([[0, 1],
       [2, 3],
       [4, 5]])
>>> print(np.mean(a,axis=0))
[2. 3.]
>>> print(np.mean(a,axis=0, keepdims=True))
[[2. 3.]]
>>> print(np.mean(a,axis=1))
[0.5 2.5 4.5]
>>> print(np.mean(a,axis=1, keepdims=True))
[[0.5]
 [2.5]
 [4.5]]
>>>