플로라도의 data workout

파이썬 - 배열(array)을 flatten 시키는 방법 본문

기초 노트/Python

파이썬 - 배열(array)을 flatten 시키는 방법

플로라도 2023. 11. 4. 21:54
nested_list = [ [1,2,3], [4,5,6], [7,8,9] ]

(1) 2중 for문의 사용
flatten_list = []
for sublist in nested_list:
	for item in sublist:
    	flatten_list.append(item)
        
(2) 리스트 컴프리헨션의 사용
flatten_list = [item for sublist in nested_list for item in sublist]

(3) 넘파이 사용 (thnkas to numpy!)
import numpy as np
nested_list = np.array(nested_list)
flatten_list = nested_list.flatten.tolist()

# 넘파이의 flatten 메서드를 사용할 경우, 이중 리스트만 풀어헤치는게 아니라 3중 리스트와 같은 텐서형태도 flatten 시킨다.