Numpy - newaxis

numpy.newaxis :  차원을 1차원 증가시켜 주는 넘파이 메소드

 

 

newaxis 메소드를 사용하기 위해, 먼저 원소가 12개인 1차원 배열을 만들어 보자.

 

import numpy as np

a = np.arange(12)

 

배열 a를 출력해보면 아래와 같이 나온다.

 

 

 

 

이제 newaxis를 이용해서 a를 2차원 행벡터로 변환해보자.

 

a_row = a[np.newaxis, : ]

 

그후 a_row를 출력해보면, 1 x 12 형태의 2차원 배열로 나온다.

 

 

 

 

 

다음은 a를 2차원 열벡터로 변환해보자.

 

a_col = a[ : , np.newaxis]

 

위 코드대로 작성하면, 1 x 12 형태의 2차원 배열인 a_col을 만들 수 있다.