[Python] Numpy np.newaxis 차원 확장

np.newaxis 차원을 확장하는 numpy 함수

행 부분에 np.newaxis를 추가하느냐

열 부분에 np.newaxis를 추가하느냐에 따라

결과 값이 달라진다.

 

---

    import numpy as np

   

   # 배열 생성

    arr1 = np.array([1, 2, 3, 4], dtype=int)

    arr2 = np.array([[1, 2], [3, 4]], dtype=int)

    print(arr1)

    print(arr2)

 

        # 결과

        [1 2 3 4]

        [[1 2]

         [3 4]]

 

   # 일차원 배열

    print(arr1[np.newaxis])

    print(arr1[:, np.newaxis])

 

        # 결과

        [[1 2 3 4]]

        [[1]

         [2]

         [3]

         [4]]

 

   # 이차원 배열

    print(arr2[np.newaxis])

    print(arr2[:, np.newaxis])

 

        # 결과

        [[[1 2

           [3 4]]]

 

        [[[1 2]]

          [[3 4]]]