[Python] np.newaxis, np.expand_dims(), np.squeeze() - 차원 및 축 조절하기

오늘은 numpy로 차원을 늘리고 줄이는 방법에 대해 포스팅해보겠다. np.newaxis() np.newaxis()는 함수명 그대로 넘파이 어레이의 차원의 수를 늘려주는 역할을 한다. 차원을 늘리고자 하는 인덱스에 np.newaxis를 포함하면 해당하는 차원이 늘어난다. import numpy as np example = np.arange(10) print(example.shape) # (10,) exam_1 = example[np.newaxis,:] exam_2 = example[:,np.newaxis] print(exam_1.shape, exam_2.shape) # (1, 10) (10, 1) 또한 한 차원이 아니라 여러 차원을 한 번에 늘릴 수도 있다. exam_3 = example[np.newax..

원문링크 : [Python] np.newaxis, np.expand_dims(), np.squeeze() - 차원 및 축 조절하기