Python
Numpy Mean Axis - 넘파이 평균 기준(축)
J-sean
2021. 7. 24. 11:23
반응형
Numpy 배열 평균을 구할때 축(axis)의 의미를 알아보자.
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
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)
|
반응형