python中 yield 用法理解

python中 yield 用法理解

---

要知道 yield 用法，就要先了解生成器，
要了解生成器，就要知道迭代器。

首先我们见到 yield 一般都是再函数中，看起来有点像 return 。
先说结论：函数中有 yield 时，返回的就是一个生成器，生成器也是可迭代对象。

def createGenerator():
...    mylist = range(3)
...    for i in mylist:
...        yield i*i
...
>>> mygenerator = createGenerator() # 创建生成器对象
>>> print(mygenerator) # mygenerator is an object!
<generator object createGenerator at 0xb7555c34>
>>> for i in mygenerator:
...     print(i)

上面代码中，调用函数时，函数里面的代码并没有执行，而是返回了一个生成器对象。$mygenerator$，当这个生成器对象的 next 方法被调用时，函数才会运行，直到遇见一个 yield 关键字，返回一个值后，函数停止运行。然后 next 方法再次被调用时，函数会从上次停止的地方继续运行，直到再次遇到 yield 关键字。不断重复。

而上面已经提到，生成器因为有 next 方法，所以是可迭代对象。
而 for i in mygenerator 时，就是不断调用生成器的 next 方法。

下面例子更直观：

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

参考：
https://blog.csdn.net/mieleizhi0522/article/details/82142856
https://stackoverflow.com/questions/231767/what-does-the-yield-keyword-do/231855#231855
https://www.cnblogs.com/wj-1314/p/8490822.html