배열의 축(axis) 이해하기

파이썬의 배열에는 축(axis)라는 개념이 있다.

파이썬의 배열은 리스트가 중첩된 것으로 표현되는데, 가장 바깥쪽 리스트가 axis=0, 안쪽으로 가면서 axis가 1씩 증가한다.

 

예를 들어, 

[ [1, 2, 3],

  [4, 5, 6] ] 

에서,

axis 0은 [1, 2, 3] → [4, 5, 6] 으로 진행되는 방향

axis 1은 1 → 2 → 3 또는 4 → 5 → 6으로 진행되는 방향

 

[ [ [ 1, 2 ],

     [ 3, 4 ] ],

   [ [ 5,  6 ],

     [ 7, 8 ] ] ]

에서,

axis 0 은 [ [ 1, 2 ], [ 3, 4 ] ] → [ [ 5, 6 ], [ 7, 8 ] ] 로 진행되는 방향

axis 1 은 [ 1, 2 ] → [ 3, 4 ] 또는 [ 5, 6 ] → [ 7, 8 ] 로 진행되는 방향

axis 2 는 1 → 2 또는 3 → 4 또는 5 → 6 또는 7 → 8 로 진행되는 방향