코딩하는 임초얀

Story 04. 리스트 컴프리헨션 본문

윤성우의 열혈 파이썬 중급편

Story 04. 리스트 컴프리헨션

초얀 2022. 7. 30. 19:54

이전 글 >> Story 03. 깊은 복사와 얕은 복사


리스트 생성 방법

리스트 컴프리헨션을 사용하여 다음과 같이 리스트를 생성할 수 있다.

>>> r1 = [1, 2, 3, 4, 5]

>>> r2 = [x * 2 for x in r1]
>>> r3 = [x + 10 for x in r1]

조건 필터 추가하기

>>> r1 = [1, 2, 3, 4, 5]
>>> r2 = [x * 2 for x in r1 if x % 2]
>>> r2
[2, 6, 10]

리스트 컴프리헨션에 for 한 번 더 들어가는 경우

>>> r1 = ['s', 't']
>>> r2 = ['1', '2', '3']
>>> r3 = [i+j for i in r1 for j in r2]
>>> r3
['s1', 's2', 's3', 't1', 't2', 't3']

이중 for 루프에 조건 필터 추가

>>> r = [n*m for n in range(2, 6) for m in range(1, 6) if (n*m)%2]
>>> r
[3, 9, 15, 5, 15, 25]
Comments