Python中的yield是什么意思

• 2019 年 10 月 6 日
• 筆記

要理解yield，先理解几个概念

可迭代的（Iterables）

从list中挨个读取item称为迭代, 如下例所示。

>>> mylist = [1, 2, 3]  >>> for i in mylist:  ...    print(i)  1  2  3

mylist是可迭代的。再来个复杂一点的例子

>>> mylist = [x*x for x in range(3)]  >>> for i in mylist:  ...    print(i)  0  1  4

生成器(Generators)

生成器是一个迭代子，但是只能迭代一次。因为它们不把数值存储在内存中，而是即时生成它们。

>>> mygenerator = (x*x for x in range(3))  >>> for i in mygenerator:  ...    print(i)  0  1  4

与上面的差别是生成器用了()，而不是[]。 不过for循环只能执行一次。执行完之后，mygenerator里面的值就变空了。

Yield

yield关键字很像return，所不同的是，它返回的是一个生成器。

>>> def createGenerator():  ...    mylist = range(3)  ...    for i in mylist:  ...        yield i*i*i  ...  >>> mygenerator = createGenerator() # 创建一个generator  >>> print(mygenerator) # mygenerator 是一个对象  <generator object createGenerator at 0xb7555c34>  >>> for i in mygenerator:  ...     print(i)  0  1  8

当你调用这个函数的时候，函数内容并没有执行，而是返回一个生成器对象。然后，你再用for去遍历这个生成器。