[Python] 리스트 평탄화 - 내포 활용

리스트가 중첩되어 있을 때, 내포를 활용해서 아래와 같이 각 리스트의 요소만 뽑아낼 수 있다.

 

* for문을 2번 사용하면 되고, 뒷쪽에 더 안쪽 요소에 접근하는 for문을 사용하면 된다.

nested_list = [ [1,2], [3,4,5], [6,7,8,9] ]

flatten_list = [ element for sub_list in nested_list for element in sub_list ]
# [1, 2, 3, 4, 5, 6, 7, 8, 9]

# sub_list는 [1,2] / [3,4,5] / [6,7,8,9]

# 참고) 행을 띄워서 적어도 됨
# flatten_list = [ element 
#                 for sub_list in nested_list
#                     for element in sub_list ]