본문 바로가기

Python

Python: 중첩 list comprehension # 이중 컴프리헨션 # itertools # chain

728x90
반응형

리스트 안에 리스트가 담긴 형태,

다른 표현으로는,  2차원 행렬을 하나의 리스트로 담고자 할때

다음과 같이 사용 가능

 

test_list = [[a,b], [c,d], [e,f]]

new_list = [j for i in test_list for j in i]

 

itertools 라이브러리의 chain.from_iterbale()을 통해서도 가능

from itertools import chain

test_list = [[a,b], [c,d], [e,f]]

new_list = list(chain.from_iterable(test_list))

결과: [a,b,c,d,e,f]

728x90
반응형