[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() 함수는 기존 객체 자체를 수정해 버린다는 점이 차이점입니다.
끝.
댓글
댓글 쓰기