numpy flatten() 함수, 1차원 배열로 변환(평탄화)

numpy의 flatten() 함수는 다차원 배열을 1차원 배열로 변환, 즉 평탄화하는 함수입니다.

2차원 배열을 1차원으로 변환

아래와 같이 2차원 배열에 대해서 flatten()을 호출하면 1차원 배열로 평탄화된 배열을 리턴합니다.

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.flatten())

Output:

[1 2 3 4 5 6]

3차원 배열을 1차원으로 변환

3차원 배열도 flatten()을 사용하여 1차원 배열로 변환할 수 있습니다. 또한, 3차원 이상의 N차원의 배열에 대해서 평탄화 할 수 있습니다.

import numpy as np

arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(arr.flatten())

Output:

[1 2 3 4 5 6 7 8]

2차원 배열을 1차원으로 변환하고 오름차순 정렬

다음은 2차원 배열을 1차원 배열로 평탄화하고, 1차원 배열을 오름차순으로 정렬하는 예제입니다.

import numpy as np

arr = np.array([[3, 2, 1], [6, 5, 4]])
print(np.sort(arr.flatten()))

Output:

[1 2 3 4 5 6]