Studies/윤성우의 열혈 파이썬 중급편
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]