[정리] Numpy ② : squeeze

b = np.array(range(1, 13, 2)).reshape(2, 3, 1) # 예상 2행 3열

 

이번에는 다차원 배열로 실습을 진행해봤는데 배열 생성 결과 이해하는 데에 한참 걸렸다. reshape 인자 순서대로 (행, 열, 차원) 인줄 알았는데? 그것이? 아니었습니다!

 

실행 결과

 

numpy array는 다른 언어에서의 배열처럼 요소 간 콤마가 안 찍히기 때문에 결과 보자마자 바로 와닿지가 않았다. 일단 reshape는 (차원, 행, 열) 크기의 다차원 배열을 만든다. → reshape(2, 3, 1)은 3행 1열의 배열을 2개 쌓았다는 뜻

 

reshape(2, 3, 1)

 

squeeze

# axis default: None, 원하는 축 지정 가능
b_squeeze = b.squeeze()

 

배열에서 길이가 1인 축을 제거한다. (2, 3, 1) 3차원에서 1을 제거했으니까 2행 3열의 2차원 배열이 되고 b_squeeze에서 대괄호 한 쌍이 줄어든 것을 확인할 수 있다.

 

실행 결과

 

인자로 제거할 축을 지정할 수 있는데 길이가 1이 아닌 축을 넣으면 에러가 발생한다.

 

위 예시에서 axis=1의 길이는 3이기 때문에 에러 발생