[Python numpy]두 개의 numpy 배열을 인덱스 방식으로 연결

💡 numpy 패키지

numpy는 과학 계산을 위한 라이브러리로서 다차원 배열을 처리하는데 필요한 여러 유용한 기능을 제공하고 있다.

numpy는 pip을 사용하여 아래와 같이 간단히 설치할 수 있다.

$ pip install numpy

💡 불러오기 

import numpy as np

💡 이전 코드 

a = np.array([[0,1,2,3,4],
              [0,1,2,3,4],
              [0,1,2,3,4]
              ...
              ])

b = np.array([[0,0,0,0,0],
              [1,1,1,1,1],
              [2,2,2,2,2],
              ...
              ])

이렇게 a, b로 나누면서 np를 불러오는 경우가 있다.

개발자로서 매우 좋지 않은 방식이며 다음과 같이 코드를 깔끔하게 불러올 수 있다.

 

💡 변경 코드 

result = np.stack((a,b), axis=2)
array([[[0, 0],
        [1, 0],
        [2, 0],
        [3, 0],
        [4, 0]],

       [[0, 1],
        [1, 1],
        [2, 1],
        [3, 1],
        [4, 1]],

       [[0, 2],
        [1, 2],
        [2, 2],
        [3, 2],
        [4, 2]]])

💡 실행 화면