-
리스트 컴프리헨션은 기본 사용법뿐만 아니라 다중 루프로 지원함
- ex: 행렬을 모든 셀이 포함된 평평한 리스트 하나로 간략화, for를 두개 사용한 리스트 컴프리헨션 (왼쪽 > 오른쪽 순서)
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] flat = [x for row in matrix for x in row] print(flat) >>> [1, 2, 3, 4, 5, 6, 7, 8, 9]
- 입력 리스트의 레이아웃을 두 레벨로 중복해서 구성(매트릭스 제곱 구하기)
squared = [[x**2 for x in row] for row in matrix] print(squared) >>> [[1, 4, 9], [16, 25, 36], [49, 64, 81]]
- 위 표현식을 다른 루프에 넣는다면, 리스트 컴프리헨션이 여러 줄로 구분해야 할 정도로 길어진다.
my_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] flat = [x for sublist1 in my_list for sublist2 in sublist1 for x in sublist2]
-
이번엔 일반 루프문으로 같은 결과를 만들어보자. 이 비전은 들여쓰기를 사용해서 리스트 컴프리헨션보다 이해하기 쉽다.
flat = []
for sublist1 in my_lists:
for sublist2 in sublist1:
flat.extend(sublist2)-
리스트 컴프리헨션도 다중 if 조건을 지원한다.
- 같은 루프 레벨에 여러조건이 있으면 암시적인
and표현식이 된다 - ex: 숫자로 구성된 리스트에서 4보다 큰 짝수 값만 가지고 오는 것
a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] b = [x for x in a if x > 4 if x % 2 == 0] c = [x for x in a if x > 4 and x % 2 == 0]
- 같은 루프 레벨에 여러조건이 있으면 암시적인
-
조건은 루프의 각 레벨에서 for 표현식 뒤에 설정 가능
- ex: 행렬에서 row의 합이 10 이상이고 3으로 나누어 떨어지는 셀을 구하자
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] filtered = [[x for x in raw if x % 3 ==0] for raw in matrix if sum(row) >= 10] print(filtered) >>> [[6], [9]]
- 리스트 컴프리헨션으로 표현하면 간단하지만 이해하기가 매우 어렵다.
-
이런 리스트 컴프리헨션은 피하라고 강력히 말하고 싶음
-
이런 코드는 다른 사람들이 이해하기가 매우 어렵다
-
몇 줄을 절약한 장점이 나중에 겪을 어려움보다 크지않음
-
리스트 컴프리헨션을 사용할 때는 표현식이 두 개를 넘어가면 피하는 게 좋다.
- 조건 두개, 루프 두개, 혹은 조건 한 개와 루프 한 개 정도면 됨
- 이것보다 복잡해지면 일반적인 if 문과 for 문을 사용하고 헬퍼 함수를 작성
- 리스트 컴프리헨션은 다중 루프와 루프 레벨별 다중 조건을 지원
- 표현식이 두 개가 넘게 들어 있는 리스트 컴프리헨션은 이해하기 매우 어려우므로 피해야 한다.