python中yield关键字怎么理解?

本文简短翻译自链接 StackOverflow
在理解yield之前,你必须先知道iterable(迭代器)和generator(生成器)。

iterable

当你创建一个list,你能一个接一个读取list的每一项。能一个接一个读取每一项的就是迭代器。

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

mylist是一个iterable,当你用列表推导式生成一个列表,也是iterable

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

一切能用for.......in.......的都是迭代器,例如lists,strings,files。。。

generator

generator是一种只能迭代一次的iterable。生成器不会再内存中储存值,它只会在程序运行的时候生成值。

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

它和上面的不同仅仅是把`[]替换成()。生成器只能使用一次。

yield

yield和return的用法相似,只是,使用yield函数将返回一个生成器。

>>> def createGenerator():
...    mylist = range(3)
...    for i in mylist:
...        yield i*i
...
>>> mygenerator = createGenerator() # create a generator
>>> print(mygenerator) # mygenerator is an object!
<generator object createGenerator at 0xb7555c34>
>>> for i in mygenerator:
...     print(i)
0
1
4

当你需要使用大量只需要使用一次的数据时,你就可以使用yield。
你需要知道的时,当你调用createGenerator()这个函数时,函数里面的代码并不会运行,函数会返回一个生成器对象。
每次对生成器对象用for,函数里面的代码都会运行一次。
以下是理解的关键部分:

for的每次循环,函数里面的代码都会执行,直到遇到yield,然后把值返回。当循环到头了或者函数不满足"if...else"条件时,生成器为空。

猜你喜欢

转载自blog.51cto.com/13592676/2285571
今日推荐