[Numpy] Flatten 기능 및 사용법 정리

numpy.ndarray.flatten

이 기능은 주어진 array를 1차원, 즉 쉽게 생각하면 한줄로 쭉 늘어놓는 함수인데요

 

 

ndarray.flatten(order='C')

세부 파라미터 order 중 가장 유용한 'C', 'F' 옵션에 대해서 더 알아보겠습니다

 

'C'   => 행단위로 추출

'F'   => 열단위로 추출

 

즉, 아래와 같은 array a를 flatten할 때 default인 'C' 행 기준으로 나열할 수도 있고

'F' 열 단위로 나열할 수도 있습니다

 

a = np.array([[1,2], [3,4]])
a.flatten()
array([1, 2, 3, 4])
a.flatten('F')
array([1, 3, 2, 4])

Source: https://numpy.org/doc/stable/reference/generated/numpy.ndarray.flatten.html