AI/Numpy
[Numpy] axis란 무엇일까?
sangwonYoon
2023. 3. 8. 01:25
numpy의 함수들의 인자로 자주 등장하는 axis의 개념에 대해서 알아보자.
axis
numpy의 함수를 실행할 때 기준이 되는 dimension 축을 의미한다.
2차원 배열을 기준으로 axis = 0이면, 같은 열에서 행 방향으로 연산을 적용하고, axis = 1이면, 같은 행에서 열 방향으로 연산을 적용한다.
test_array = np.arange(1, 13).reshape(3, 4)
print(test_array)
# 출력:
# [[ 1 2 3 4]
# [ 5 6 7 8]
# [ 9 10 11 12]]
print(test_array.sum(axis=1)) # 같은 행에서 열 방향으로 값들을 더한다.
# 출력: [10 26 42]
print(test_array.sum(axis=0)) # 같은 열에서 행 방향으로 값들을 더한다.
# 출력: [15 18 21 24]

