티스토리 뷰


리스트를 생성하는 생성하는 컴프리헨션을


리스트 컴프리헨션이라고 한다.



ns = []
for n in range(10):
ns.append(n)


위의 코드를 컴프리헨션으로 작성


ns = [n for n in range(10)]



조건문을 적용할 수도 있다.


ns = []
for n in range(10):
if n % 2:
ns.append(n)


위를 컴프리헨션으로 작성하면


ns = [n for n in range(10) if n % 2]



다중 for 문과 if 문을 적용할수도 있다.


items = []
for c in ["a", "b", "c"]:
for j in range(10):
if j < 5 and j % 2:
items.append((c, j))


컴프리헨션으로 작성하면


items = [(c, j) for c in ["a", "b", "c"] for j in range(10) if j < 5 if j % 2]


for 문은 맨왼쪽부터 최상위 for 문으로 적용된다.


댓글