python之mean函数,计算多维

mean() 函数定义： 
numpy.mean(a, axis, dtype, out，keepdims )

mean()函数功能：求取均值 
经常操作的参数为axis，以m * n矩阵举例：

• axis 不设置值，对 m*n 个数求均值，返回一个实数
• axis = 0：压缩行，对各列求均值，返回 1* n 矩阵
• axis =1 ：压缩列，对各行求均值，返回 m *1 矩阵

 

arr = np.array([[1,2,3],
                [4,5,6]])

print(arr.mean()) #计算所有数的平均值
print(arr.mean(0)) #计算列上平均值
print(arr.mean(1)) #计算行上平均值

多维的转化:

arr = np.array([
                [[1,2,3],
                 [4,5,6]],
                [[7,8,9],
                 [10,11,12]]
])

2*2*3
print(arr.shape)

print(arr.mean(0).shape) #(2L, 3L)
print(arr.mean(0)) # 对应列上的平均值
#记住一点 压缩行，列上平均值


print(arr.mean(1).shape)
print(arr.mean(1)) # 对应行上的平均值
'''
    [[a1,a2,a3],
     [a4,a5,a6]]

     [[b1,b2,b3],
      [b4,b5,b6]]
    axis=0 在列上的转化:
        [a1+b1, a2+b2, a3+b3],
        [a4+b4, a5+b5, a6+B6]

    axis=1 在行的的转化:
        [a1+a4, a2+a5, a3+a6],
        [b1+b4, b2+b5, b3+b6]
'''

print('---------')
print(arr.mean(1).mean(1).shape)
print(arr.mean(1).mean(1))
'''
    [a1+a2+a3+a4+a5+a6, b1+b2+b3+b4+b5+b6]
'''