Python生成器next方法和send方法区别
yield的语法规则是:
在yield这里暂停函数执行,并返回yield后面表达式的值(默认为None),直到被next()再次调用时,从上次暂停的yield代码处继续往下执行。当没有可继续next()时,抛出异常,该异常可被for循环处理。
def fib(n): a, b = 0, 1 i = 0 while i < n: yield b a, b = b, a+b i += 1 if __name__ == ‘__main__‘: f = fib(10) for item in f: print(item)
python中,含有yield关键字的对象就是一个生成器,每次调用next方法时会执行到yield后面的语句,然后返回yield后面代码块的执行结果
def foo(): bar_a = yield 1 # bar_a是语句块(yield 1)的返回值,默认为None bar_b = yield bar_a yield "最后一个值,再迭代就要报StopIteration了" f = foo() # 创建生成器,此时没有执行foo()里的任何语句 print(next(f)) # 从foo()里进入,一直执行到(yield 1)处,此时变量bar_a还没有创建 print(next(f)) # 先将语句块(yield 1)的返回值赋值个bar_a,此时bar_a的值是None。 # 然后执行到语句块(yield bar_a),bar_b也还没有被创建 print(next(f) >>>1 >>>None >>>最后一个值,再迭代就要报StopIteration了
send()
def foo(): bar_a = yield 1 bar_b = yield bar_a yield "最后一个值,再迭代就要报StopIteration了" f = foo() print(f.send(None)) # 和next(f)的效果一模一样 print(f.send("my lover")) # 覆盖yield 1语句的返回值 print(next(f)) >>>1 >>>my lover >>>最后一个值,再迭代就要报StopIteration了