[python] 파이썬 다차원 배열을 1차원 배열로 변환

파이썬에서 다차원 배열을 1차원 배열로 변경시켜 보겠습니다. 우선 아래와 같은 배열을 선언합니다.


코드는 아래와 같습니다.


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


위 배열을 아래와 같은 1차원 배열로 변경하고자 합니다.


flatten() 함수와 ravel() 함수 2가지가 있습니다. 코드는 아래와 같습니다.


import numpy as np

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

b = a.flatten()
print(b)

c = np.ravel(a)
#c = a.ravel()
print(c)


flatten() 함수는 객체의 사본을 반환하기 때문에 사본 내용을 바꾸어도 원본이 영향받지 않지만, ravel() 함수는 기존 객체 자체를 수정해 버린다는 점이 차이점입니다.


끝.




댓글