python yield学习

yield 在 Python 中被称之为 generator（生成器），yield常用于生成一个迭代器。

列表与迭代器

假设我们要实现一个函数，它需要返回一个斐波那契数列的前n个数，平常我们会写作如下

def fb(max): 
   n, a, b = 0, 0, 1 
   res = [] 
   while n < max: 
       res.append(b) 
       a, b = b, a + b 
       n = n + 1 
   return res

我们可以通过如下方式调用

for n in fb(5): 
    print n 

输出为

1
1
2
3
5

但是假如n十分大，大到fb函数创建一个列表就要耗费掉机器的所有内存时，我们便不能通过上述方法来实现。幸运的是python提供了yield关键字。yield返回的是一个迭代器，我们只需将上述fb函数稍作修改

def fb(max): 
   n, a, b = 0, 0, 1 
   #res = [] 
   while n < max: 
       #res.append(b) 
       yield b 
       a, b = b, a + b 
       n = n + 1 
   #return res

我们依旧可以通过如下方式调用

for n in fb(5): 
    print n 

输出为

1
1
2
3
5

不过此时fb已经变为了一个生成器。带有 yield 的函数不再是一个普通函数，Python 解释器会将其视为一个 生成器，调用 fb(5) 不会执行fb 函数，而是返回一个 iterable 对象。在 for 循环执行时，每次循环都会执行 fb 函数内部的代码，执行到 yield b 时，fb 函数就返回一个迭代值，即b的值，下次迭代时，代码从 yield b 的下一条语句继续执行，而函数的本地变量看起来和上次中断执行前是完全一样的，于是函数继续执行，直到再次遇到 yield。

send()与next()方法

除了使用for来迭代以外，我们还可以通过调用迭代器的next方法

我们重新创建一个十分简单的函数

#!coding: utf-8
def get_primes():
    yield 5
    yield 6
    yield 7

调用方式如下

a = get_primes()
s = a.next()
print s
s = a.next()
print s
s = a.next()
print s
s = a.next()     #此步会报错，因为已经到达函数尾
print s

输出结果如下

5
6
7
Traceback (most recent call last):
  File "test.py", line 19, in <module>
    s = a.next()
StopIteration

send()方法用于向迭代器发送数据，具体请看如下事例

def get_primes():
    m = 1            #无实际意义，仅为演示   
    s = yield 5
    print '1',s
    s = yield 6
    print '2',s
    s = yield 7
    print '3',s

a = get_primes()
s = a.next()          #等价于 a.send(None)，由于send方法是将值赋值给yield，而在函数开头并不一定会存在yield
print s               #上一步执行到yield，结束并返回5给s
m = a.send('hello')   #从上一步开始执行，即s = 'hello'(yiled 5),知道碰到下一个yield，并返回其值，为6.
print m
m = a.send('world')
print m
m = a.send('!')       #由于yield 7 后再无yield语句，故此步会报错
print m

输出结果如下

5
1 hello
6
2 world
7
3 !
Traceback (most recent call last):
  File "test.py", line 19, in <module>
    m = a.send('!')
StopIteration

从上列可以看出，send()方法用于向生成器中传递值。

总结

• generator是用来产生一系列值的
• yield则像是generator函数的返回结果
• yield唯一所做的另一件事就是保存一个generator函数的状态
• generator就是一个特殊类型的迭代器（iterator）
• 和迭代器相似，我们可以通过使用next()来从generator中获取下一个值
• 通过隐式地调用next()来忽略一些值
• 通过send()我们可以向yield传递数据

注意事项

       next()为python2.x的用法，在python3中变为__next__()

参考资料 

       https://www.ibm.com/developerworks/cn/opensource/os-cn-python-yield/

       https://www.oschina.net/translate/improve-your-python-yield-and-generators-explained?lang=chs&page=1#

第一次写博客，有什么不足还请大家指正，谢谢～