Python生成器next方法和send方法区别

Posted 2020-12-22 BSSZDS930

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了