Python中yeild的用法详解

Python中yeild的用法详解

1. 初步认识：先把yield看做return

   return:在程序中返回某个值，返回后程序就不再往下运行了

   代码：

   def foo():
        print("starting...")
        while True:
            res = yield 4
            print("res:",res)
    g = foo()
    print(next(g))
    print("*"*20)
    print(next(g))
              

   代码输出：

   starting...
    4
    ********************
    res: None
    4
              

   代码解释：
   1. 程序开始执行的时候，因为foo函数中有yield关键字，所以foo函数并不会真的执行，而是先得到一个生成器g（相当于一个对象）
   2. 直到调用next方法，foo函数正式开始执行，先执行foo函数中的print方法，然后进入while循环
   3. 程序遇到yield关键字，然后把yield想成return，return了一个4后，程序停止，并没有执行操作’赋值给res’，此时next(g)语句执行完成，所以输出的前两行（第一个是while上面的print的结果，第二个是return出的结果）是执行print(next(g))的结果
   4. 程序执行

      print("*""*20)

      ,输出20个*
   5. 再次执行

      print(next(g))

      ,此时从next程序停止的地方开始执行。执行res的赋值操作，此时赋值操作的右边是没有值的（因为刚才那个被return出去了，并没有给赋值操作的左边传参数），所以这个时候res赋值是None，所以输出res:None
   6. 程序会继续在while里执行，又一次碰到yield,这个时候同样return出4，然后程序停止，print函数输出的4就是这次return出的4

   总结：

   带yield的函数是一个生成器，而不是一个函数了，这个生成器有一个函数就是next函数，next就相当于“下一步”生成哪个数，这一次的next开始的地方是接着上一次的next停止的地方执行的，所以调用next的时候，生成器并不会从foo函数的开始执行，只是接着上一步停止的地方开始，然后遇到yield后，return出要生成的数，此步就结束。

2. 再认识：yield是生成器（generator）的一部分（带yield的函数才是真正的迭代器）

   代码：

   def foo():
        print("starting...")
        while True:
            res = yield 4
            print("res:",res)
    g = foo()
    print(next(g))
    print("*"*20)
    print(g.send(7))
              

   输出结果：

   starting...
    4
    ********************
    res: 7
    4
              

   代码解释：
   1. 程序执行g.send(7),程序会从yield关键字那一行继续向下运行,send会把7这个值赋值给res变量
   2. 由于send方法中包含next()方法，所以程序会继续向下运行执行print方法，然后再次进入while循环
   3. 程序执行再次遇到yield关键字，yield会返回后面的值，程序再次停住，直到再次调用next方法或send方法

3. 合理使用

   取0，1，2，3，… 1000：

   代码

   for n in range(1000):
        a = n
              

   此时，range(1000)就默认生成了一个含有1000个数的list，所以很占内存。

   可以用刚才的yield组合生成器进行实现:

   def foo(num):
        print("starting...")
        while num<10:
            num=num+1
            yield num
    for n in foo(0):
        print(n)
              

   也可以用xrange(1000)这个生成器实现:

   for n in range(1000):
        a=n
    注意：python3时已经没有xrange()了，在python3中range()就是xrange()