python中yield用法

参看这里


 yield在函数中的功能类似于return,不同的是yield每次返回结果之后函数并没有退出,而是 每次遇到yield关键字后返回相应结果,并保留函数当前的运行状态,等待下一次的调用。如果 一个函数需要多次循环执行一个动作,并且每次执行的结果都是需要的,这种场景很适合使用yield实现。
包含yield的函数成为一个生成器,生成器同时也是一个迭代器,支持通过next方法获取下一个值。

yield基本使用:

  1. def func():
  2. for i in range(0,3):
  3. yield i
  4. f = func()
  5. f.next()
  6. f.next()

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

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


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

  1. def func(n):
  2. for i in range(0,n):
  3. val = yield i
  4. print val
  5. f = func(10)
  6. f.next()
  7. #f.send(None)
  8. f.send(2)
  9. f.send(10)
  10. print f.next()