python生成器

本文已参与「新人创作礼」活动，一起开启掘金创作之路

• 生成器(generator)：生成器是一类特殊的迭代器。

• 生成器每次在迭代时可以返回一个或者多个值，它可以记录当前状 态。

• 生成器的创建方式：

 – 使用yield关键字。
 – 使用生成器表达式(推导式)。
 

生成器推导式

• 推导式形式：

– (推导式)：G = ( x*2 for x in range(5)

– 判断是否是生成器：

yield创建生成器

• 使用yield创建生成器：

– 通过yield创建一个产生斐波那契数列数列的生成器。

生成器取值

• 生成器可以使用next和send取值：

– next，获取下一个元素。

– send，获取下一个元素，同时可以向生成器中传递一个值。

– next等同于send(None)。

– 第一次取值时，需要使用next或者send(None)

执行过程：

– 4 - 1 - 2 - 3 - 5 - 1 - 2 …… - 6

def fib(n): 
  current = 0 
  num1, num2 = 0, 1 
  while current < n: 
    num = num1 # 1 
    num1, num2 = num2, num1+num2 # 2 
    current += 1 
    yield num # 3 
  return 'done' # 6 
g=fib(5) # 4 
next(g) # 5

• 使用yield声明函数为生成器。

• 当程序执行到yield处时，生成器函数暂停，等待下一次唤醒（next、 send）。

– 并且返回当前值。

– 和return相似，但是程序暂停而不是结束。

通过 yield 创建生成器函数：

def fib(n):
  current = 0 
  num1, num2 = 0, 
  1 while current < n: 
    num = num1 
    num1, num2 = num2, num1+num2 
    current += 1 
    yield num 
    return 'done' 
    g=fib(5) 
    while True: 
      try: 
          x = next(g) 
          print("value:%d"%x) 
      except StopIteration as e: 
          print("生成器返回值:%s"%e.value) 
          break

除了使用 next 方法以外，还可以使用 send 方法唤醒生成器，相比于 next 方法，send 在唤醒生成器时还 可以向断点处传入一个数据。

def gen(): 
  i = 0 
  while i<5: 
    temp = yield i 
    print(temp) 
    i+=1 
f = gen() 
next(f) 
>>>0 
f.send('haha') 
>>>haha 
>>>1 
next(f) 
>>>None 
>>>2