包含yield表达式的函数是特殊的函数,叫做生成器函数(generator function),被调用时将返回一个迭代器(iterator),调用时可以使用next或send(msg)。它的用法与return相似,区别在于它会记住上次迭代的状态,继续执行。
send(msg)与next()的区别在于send可以传递参数给yield表达式,这时传递的参数会作为yield表达式的值,而yield的参数是返回给调用者的值。初始调用时必须先next()或send(None),否则会报错。
举个例子:
首先生成一个迭代器f,f.next()会使生成器函数执行到yield,生成一个值然后挂起。
然后f.next()或f.send(msg)会在生成器函数内部返回值,执行到下一个yield,生成值后挂起
然后f.next()或f.send(msg)会在生成器函数内部返回值,意图执行到下一个yield,但是后面没有yield了,所以抛出异常。
使用yield可以有效简化代码,并减少空间浪费。
举个简单例子:列表中的每个元素+1
传统写法:
Python代码
- def addlist(alist):
- r = []
- for i in alist:
- r.append(i+1)
- return r
复制代码 yield写法:
Python代码
- def addlist(alist):
- for i in alist:
- yield i + 1
复制代码 当然对于这种简单的问题:
Python代码
- [i+1 for i in alist]
复制代码