[python list] 파이썬 리스트 안 리스트 펼치기_how to flatten a list of lists in python

#파이썬 리스트 펼치기 #파이썬 2d 리스트 펼치기 #python 2d list flatten #python nested list flatten 

 

다음과 같이 2D list를 1D list로 전환하는 방식을 'flattening'이라고 한다

1. itertools의 chain 함수 사용하기

import itertools

List_2D = [['a','b','c'],[1,2,3],['d',4,'e']] #List to be flattened

List_flat = list(itertools.chain(*List_2D))

print("Original List:",List_2D)
print("Flattened List:",List_flat)
# print문 출력결과
# Original List: [['a', 'b', 'c'], [1, 2, 3], ['d', 4, 'e']]
# Flattened List: ['a', 'b', 'c', 1, 2, 3, 'd', 4, 'e']

 

2. functools의 reduce 함수와 operator의 iconcat 함수 사용하기 (1번보다 빠름)

import functools
import operator

lst_2D = [['a','b','c'],[1,2,3],['d',4,'e']] #List to be flattened

lst_flat = functools.reduce(operator.iconcat, lst_2D, [])

print("Original List:",lst_2D)
print("Flattened List:",lst_flat)
# print 출력결과
# Original List: [['a', 'b', 'c'], [1, 2, 3], ['d', 4, 'e']]
# Flattened List: ['a', 'b', 'c', 1, 2, 3, 'd', 4, 'e']

 

 

 

reference : www.educative.io/edpresso/how-to-flatten-a-list-of-lists-in-python