Numpy Mean Axis - 넘파이 평균 기준(축)

Numpy 배열 평균을 구할때 축(axis)의 의미를 알아보자.

 

data = np.array([[[1, 1, 1, 1],
                [1, 1, 1, 1],
                [1, 1, 1, 1]],
 
                [[2, 2, 2, 2],
                [2, 2, 2, 2],
                [2, 2, 2, 2]]])
 
# 2X3X4 행렬 (3행4열2그룹)
 
print(data)
print(data.shape) # (2, 3, 4)
 
print(np.mean(data, axis=0)) # 각 그룹의 같은 원소끼리 평균(3X4)
print(np.mean(data, axis=1)) # 각 그룹의 열 평균(2X4)
print(np.mean(data, axis=2)) # 각 그룹의 행 평균(2X3)
 
print(np.mean(data, axis=(1, 2))) # 각 그룹의 평균(1X2)