본문 바로가기
머신러닝을 위한 파이썬 기초/넘파이

파이썬 넘파이 1차원 열벡터로 바꾸는 방법

by bigpicture 2023. 10. 26.
반응형

reshape 함수를 사용합니다. 아래와 같이 사용합니다. 

import numpy as np

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

>>> a.reshape((-1,1))
array([[1],
       [2],
       [3],
       [4],
       [5],
       [6]])


-1이 뭔지 궁금해하시는 분들이 많은데 -1은 열벡터로 만들겠다는 뜻입니다. -1,1 은 열벡터로 만들고 원소 하나씩을 갖게 하겠다는 의미입니다. -1,2 는 열벡터로 만들고 원소 두개씩을 갖게 합니다. 

>>> a.reshape((-1,2))
array([[1, 2],
       [3, 4],
       [5, 6]])
반응형

댓글