[Numpy] numpy.mean 의 Axis(축) 에 대하여

import numpy as np

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

# 2 x 3 x 4 행렬 (2채널 3행 4열)

print(data.shape, end="\n\n")   #(2, 3, 4)

print(np.mean(data, axis=0),end="\n\n") # 각 채널 같은 위치 원소끼리 평균 (3x4)
print(np.mean(data, axis=1),end="\n\n") # 각 채널의 열끼리의 평균 (2x4)
print(np.mean(data, axis=2),end="\n\n") # 각 채널의 행끼리의 평균 (2x3)

print(np.mean(data, axis=(1,2))) # 각 채널의 평균 (1x2)

 

출력 :