전문가를 위한 파이썬(2판)

692

4

부

제어 흐름

제너레이터 표현식을 사용하면

Sentence

클래스의 코드를 더 간단히 만들 수 있다. [예제

17

-

10

]을 보자.

예제

17-10

sentence

_

genexp

.

py

: 제너레이터 표현식을 사용한

Sentence

import re

import reprlib

RE_WORD = re.compile(r’\w+’)

class Sentence:

def __init__(self, text):

self.text = text

def __repr__(self):

return f’Sentence({reprlib.repr(self.text)})’

def __iter__(self):

return (match.group() for match in RE_WORD.finditer(self.text))

[예제

17

-

8

]과

__iter__( )

메서드만 다르다. 여기서는 제너레이터 함수가 아니라 (

yield

문

이 없다 ), 제너레이터를 생성해 반환하는 제너레이터 표현식을 사용한다. 실행 결과는 [예제

17

-

8

]과 똑같다.

__iter__( )

메서드의 호출자가 제너레이터 객체를 받는다.

제너레이터 표현식은 편리 구문이다. 제너레이터 함수로 대체할 수 있지만, 때로는 제너레이터

표현식이 더 편리하다. 다음 절에서는 제너레이터 표현식을 사용하는 방법을 알아보자.

17

.