[Python] 이중 리스트 해제하고 1차원 리스트로 만들기

이중 리스트란 쉽게 말해 리스트 안에 리스트가 있는 것을 말합니다. 

 

sample_li = [[1, 2, 3], [4, 5, 6]]

 

이러한 리스트의 경우, 리스트 안의 특정값을 찾거나 카운트하는 과정이 번거로워집니다. 그렇기 때문에 이러한 이중 리스트를 해제하고 그 안의 값들만 모아 1차원 리스트로 만드는 방법을 알아두면 편리합니다.   

 

# 라이브러리 불러오기
import itertools

# 샘플 리스트 만들기
sample_li = [[1,2,3], [4,5,6]]

# 이중 리스트 해제하기
sample_li = list(itertools.chain(*sample_li))

>> [1,2,3,4,5,6]

 

리스트 해제 시에는 itertools라는 라이브러리를 사용합니다. 이 라이브러리는 반복되는 데이터를 처리하는 기능을 가지고 있으며, 여러 클래스를 제공합니다(대표적으로 사용되는 클래스로 주어진 수들의 순열과 조합을 알려주는 permutations와 combinations가 있음).

 

하지만 위에서 보다시피 이중 리스트를 해제하는 데에도 요긴하게 쓰이므로 알아두면 좋습니다.