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

import numpy as np

#2차원 배열을 1차원으로 변환
arr = np.array([[1,2,3],[4,5,6]])
print(arr.flatten())

#3차원 배열을 1차원을 변환
arr = np.array([[[1,2],[3,4]],[[5,6],[7,8]]]) #2행2열
print(arr.flatten())

#2차원 배열을 1차원으로 변환하고 오름차순 정렬
arr = np.array([[3,2,1],[6,5,4]])
print(arr.flatten())
print(np.sort(arr.flatten()))

 

참고 및 출처:

https://codechacha.com/ko/python-numpy-flatten/

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