컴프리헨션과 제너레이터¶

파이썬에서는 컴프리헨션이라는 구문을 사용할 수 있다.

컴프리헨션 코딩 스타일은 제너레이터를 사용하는 함수로 확장할 수 있다.

제너레이터는 함수가 점진적으로 반환하는 값으로 이뤄지는 스트림을 만들어준다.

이터레이터를 사용할 수 있는 곳이라면 어디에서나 제너레이터 함수를 호출한 결과를 사용할 수 있다.

27. map과 filter 대신 컴프리헨션을 사용하라¶

리스트 컴프리헨션!

In [1]:

a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
squares = []
for x in a:
    squares.append(x**2)
print(squares)

[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

In [2]:

squares = [x**2 for x in a]
print(squares)

[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

In [3]:

alt = map(lambda x: x ** 2, a)

In [5]:

even_squares = [x**2 for x in a if x % 2 == 0]
print(even_squares)

[4, 16, 36, 64, 100]

In [6]:

alt = map(lambda x : x**2, filter(lambda x: x % 2 == 0, a))

In [7]:

even_squares

Out[7]:

[4, 16, 36, 64, 100]

In [8]:

even_squares_dict = {x: x**2 for x in a if x % 2 == 0}
threes_cubed_set = {x**3 for x in a if x % 3 == 0}
print(even_squares_dict)
print(threes_cubed_set)

{2: 4, 4: 16, 6: 36, 8: 64, 10: 100}
{216, 729, 27}

In [9]:

alt_dict = dict(map(lambda x: (x, x**2), filter(lambda x: x % 2 == 0, a)))

In [10]:

alt_set = set(map(lambda x: (x, x**3), filter(lambda x: x % 3 == 0, a)))

기억해야 할 내용¶

리스트 컴프리헨션은 lambda 식을 사용하지 않기 때문에 같은 일을 하는 map과 filter 내장 함수를 사용하는 것보다 더 명확하다.
리스트 컴프리헨션을 사용하면 쉽게 입력 리스트의 원소를 건너뛸 수 있다. 하지만 map을 사용하는 경우에는 filter의 도움을 받아야만 한다.
딕셔너리와 집합도 컴프리헨션으로 생성할 수 있다.