[Numpy] axis란 무엇일까?

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]

2차원 배열에서의 axis

3차원 배열에서의 axis