[BW 28] 컴프리헨션 내부에 제어 하위 식을 세 개 이상 사용하지 말라

컴프리헨션은 루프를 여러 수준으로 내포하도록 허용한다.

예를 들어 리스트 안에 리스트가 들어 있는 형태로 정의한 행렬을 단일 리스트로 단순화하는 예제를 살펴보자.

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]

 

또한, 컴프리헨션에서 다중 루프를 사용하는 예를 살펴보자.

matrix = [[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]]

 

하지만 많은 루프 또는 조건문을 사용해 컴프리헨션을 작성할 때는 다른 대안이 더 명확해 보일 수 있다.

# 컴프리헨션을 사용하는 경우
matrix = [[[1, 2, 3], [4, 5, 6], [7, 8, 9]], [[10, 11, 12], [13, 14, 15], [16, 17, 18]]]
flat = [x for sublist1 in matrix for sublist2 in sublist1 for x in sublist2]

# 다른 대안
flat = []
matrix = [[[1, 2, 3], [4, 5, 6], [7, 8, 9]], [[10, 11, 12], [13, 14, 15], [16, 17, 18]]]
flat = []
for sublist1 in matrix:
    for sublist2 in sublist1:
        flat.extend(sublist2)
        
print(flat)

>>>
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18]