带有yield的函数在Python中如何使用

yield在函数中的功能类似于return,不同的是yield每次返回结果之后函数并没有退出,而是每次遇到yield关键字后返回相应结果,并保留函数当前的运行状态,等待下一次的调用。如果一个函数需要多次循环执行一个动作,并且每次执行的结果都是需要的,这种场景很适合使用yield实现。

包含yield的函数成为一个生成器,生成器同时也是一个迭代器,支持通过next方法获取下一个值。

yield基本使用:

def func():
    for i in range(0,3):
        yield i
 
f = func()
f.next()
f.next()

对于生成器,当调用函数next时,将获取生成器yield后边表达式的值;

当执行完最后一次循环后,结束yield语句,生成器会抛出StopIteration异常;

除了next函数,生成器还支持send函数。该函数可以向生成器传递参数。

def func(n):
    for i in range(0,n):
        val = yield i        
        print val
 
f = func(10)
f.next()
#f.send(None)
f.send(2)
f.send(10)
print f.next()

更多学习内容,请点击Python学习网。