Python中yield的用法

来源:互联网 发布:色戒 梁朝伟 知乎 编辑:程序博客网 时间:2024/05/21 06:34

yield是生成的意思,但是在python中则是作为生成器理解,生成器的用处主要可以迭代,这样简化了很多运算模型(还不是很了解是如何简化的)。
yield是一个表达式,是有返回值的.
当一个函数中含有yield时,它不再是一个普通的函数,而是一个生成器.当该函数被调用时不会自动执行,而是暂停,见第一个例子:
例1:
[python] view plain copy

def mygenerator():
… print ‘start…’
… yield 5

mygenerator() //在此处调用,并没有打印出start…说明存在yield的函数没有被运行,即暂停

mygenerator().next() //调用next()即可让函数运行.
start…
5

如一个函数中出现多个yield则next()会停止在下一个yield前,见例2:
例2:
[python] view plain copy
def fun2():
… print ‘first’
… yield 5
… print ‘second’
… yield 23
… print ‘end…’

g1 = fun2()
g1.next() //第一次运行,暂停在yield 5
first
5
g1.next() //第二次运行,暂停在yield 23
second
23
g1.next() //第三次运行,由于之后没有yield,再次next()就会抛出错误
end…
Traceback (most recent call last):
File “”, line 1, in
StopIteration

为什么yield 5会输出5,yield 23会输出23?
我们猜测可能是因为yield是表达式,存在返回值.
那么这是否可以认为yield 5的返回值一定是5吗?实际上并不是这样,这个与send函数存在一定的关系,这个函数实质上与next()是相似的,区别是send是传递yield表达式的值进去,而next不能传递特定的值,只能传递None进去,因此可以认为g.next()和g.send(None)是相同的。见例3:
例3:
[python] view plain copy
def fun():
… print ‘start…’
… m = yield 5
… print m
… print ‘middle…’
… d = yield 12
… print d
… print ‘end…’

m = fun() //创建一个对象
m.next() //会使函数执行到下一个yield前
start…
5
m.send(‘message’) //利用send()传递值
message //send()传递进来的
middle…
12
m.next()
None //可见next()返回值为空
end…
Traceback (most recent call last):
File “”, line 1, in
StopIteration