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])
>>> 

矩阵相减：n维矩阵减1维矩阵，会将n维矩阵的每一个维度都减去该一维矩阵

>>> a
array([[0, 1, 2, 3],
       [0, 1, 2, 3]])
>>> b=np.mean(a,axis=0)
>>> b
array([0., 1., 2., 3.])
>>> c=a-b
>>> c
array([[0., 0., 0., 0.],
       [0., 0., 0., 0.]])