[python] 파이썬에서 생성기 이해

참고 :이 게시물은 Python 3.x 구문을 가정합니다. ^†

발전기는 단순히 당신이 호출 할 수있는 객체를 반환하는 함수입니다 next그것이 제기 할 때까지 모든 호출 것이, 어떤 값을 반환 등 StopIteration모든 값이 생성 된 것을 신호, 예외를. 이러한 객체를 반복 자라고합니다 .

일반 함수 return는 Java와 마찬가지로을 사용하여 단일 값을 반환합니다 . 그러나 파이썬에는이라는 대안이 yield있습니다. yield함수의 어느 곳에서나 사용 하면 생성기가됩니다. 이 코드를 준수하십시오.

>>> def myGen(n):
...     yield n
...     yield n + 1
... 
>>> g = myGen(6)
>>> next(g)
6
>>> next(g)
7
>>> next(g)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

당신이 볼 수 있듯이, myGen(n)산출하는 기능입니다 n및 n + 1. 모든 next값 을 산출 할 때까지 모든 호출 은 단일 값을 산출합니다. for루프 next는 백그라운드에서 호출 되므로 다음과 같습니다.

>>> for n in myGen(6):
...     print(n)
... 
6
7

마찬가지로 특정 일반적인 유형의 생성기를 간결하게 설명하는 수단을 제공하는 생성기 표현식 이 있습니다 .

>>> g = (n for n in range(3, 5))
>>> next(g)
3
>>> next(g)
4
>>> next(g)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

생성기 표현식은 목록 이해 와 매우 비슷 합니다 .

>>> lc = [n for n in range(3, 5)]
>>> lc
[3, 4]

생성기 객체는 한 번 생성 되지만 해당 코드는 한 번에 실행 되지 않습니다 . next실제로 코드를 실행 (일부)하는 호출 만 가능합니다 . yield명령문에 도달 하면 생성기에서 코드 실행이 중지 되고 값이 리턴됩니다. 다음에 호출 next하면 생성기가 마지막 이후에 남아있는 상태에서 실행이 계속됩니다 yield. 이는 정규 함수와의 근본적인 차이점입니다. 즉, “상단”에서 실행을 시작하고 값을 반환하면 상태를 버립니다.

이 주제에 관해 할 말이 더 있습니다. 예를 들어 send생성기로 다시 데이터를 보내는 것이 가능합니다 ( 참조 ). 그러나 그것은 발전기의 기본 개념을 이해할 때까지 살펴 보지 말 것을 제안하는 것입니다.

이제 물어볼 수 있습니다 : 왜 발전기를 사용합니까? 몇 가지 좋은 이유가 있습니다.

• 특정 개념은 생성기를 사용하여 훨씬 간결하게 설명 할 수 있습니다.
• 값 목록을 반환하는 함수를 만드는 대신 값을 즉시 생성하는 생성기를 작성할 수 있습니다. 이것은리스트를 구성 할 필요가 없다는 것을 의미하며, 결과 코드는 메모리 효율성이 높다는 것을 의미합니다. 이러한 방식으로 메모리에 맞추기에는 너무 큰 데이터 스트림을 설명 할 수도 있습니다.

• 생성기는 무한 스트림 을 자연스럽게 표현할 수있는 방법을 제공 합니다. 예를 들어 피보나치 수를 고려하십시오 .

  >>> def fib():
  ...     a, b = 0, 1
  ...     while True:
  ...         yield a
  ...         a, b = b, a + b
  ... 
  >>> import itertools
  >>> list(itertools.islice(fib(), 10))
  [0, 1, 1, 2, 3, 5, 8, 13, 21, 34]
  

  이 코드는 itertools.islice무한 스트림에서 유한 요소 수를 가져 오는 데 사용 됩니다. itertools고급 생성기를 작성하기위한 필수 도구이므로 모듈 의 기능을 잘 살펴 보는 것이 좋습니다 .

  ^† Python <= 2.6 정보 : 위의 예 에서 지정된 객체 next의 메서드를 호출하는 함수입니다 __next__. 파이썬 <2.6 용도, 즉 하나는 약간 다른 기술 = o.next()대신이 next(o). Python 2.7에는 next()호출이 .next있으므로 2.7 에서 다음을 사용할 필요가 없습니다.

>>> g = (n for n in range(3, 5))
>>> g.next()
3

답변

>>> def myGenerator():
...     yield 'These'
...     yield 'words'
...     yield 'come'
...     yield 'one'
...     yield 'at'
...     yield 'a'
...     yield 'time'

>>> myGeneratorInstance = myGenerator()
>>> next(myGeneratorInstance)
These
>>> next(myGeneratorInstance)
words

>>> for word in myGeneratorInstance:
...     print word
These
words
come
one
at
a
time

>>> from time import gmtime, strftime
>>> def myGen():
...     while True:
...         yield strftime("%a, %d %b %Y %H:%M:%S +0000", gmtime())
>>> myGeneratorInstance = myGen()
>>> next(myGeneratorInstance)
Thu, 28 Jun 2001 14:17:15 +0000
>>> next(myGeneratorInstance)
Thu, 28 Jun 2001 14:18:02 +0000   

>>> g = (x for x in range(10))
>>> g
<generator object <genexpr> at 0x7fac1c1e6aa0>
>>> g.next()
0
>>> g.next()
1
>>> g.next()
2
>>> list(g)   # force iterating the rest
[3, 4, 5, 6, 7, 8, 9]
>>> g.next()  # iterator is at the end; calling next again will throw
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

>>> def infinite_gen():
...     n = 0
...     while True:
...         yield n
...         n = n + 1
...
>>> g = infinite_gen()
>>> g.next()
0
>>> g.next()
1
>>> g.next()
2
>>> g.next()
3
...

#! /usr/bin/python
def  mygen(n):
    x = 0
    while x < n:
        x = x + 1
        if x % 3 == 0:
            yield x

for a in mygen(100):
    print a