生成器概念;
生成器: 函数中有yield 或者 yield from , 或者生成器表达式(x for x in range(5)) 都会被编译成生成器.
生成器内部实现了迭代协议,即__iter__ , __next__ 所以可以用于迭代.
生成器不会像列表一样一次在内存中展开所有的值, 而是拿一次计算一次.
生成器是一个对象, 因此使用生成器需要调用一下函数来创建一个生成器对象,然后进行迭代
关于yield:
生成器的核心是 yield , 一旦运行到 yield 就暂停, 如要继续生成器内的代码, 需要外部驱动( next / send),
next / send 都会先调用生成器__iter__ , 然后每次__next__
因此就目前为止先对生成器有一个 可暂停的印象即可;
同时需要注意的是 , 生成器函数一旦被调用 ,此生成器对象并不会马上执行代码 (跟函数的本质区别),需要你用next 来驱动一下,
生成器的结束方式也与普通不一样, 不论在生成器内部何时return / 执行到末尾 都会产生一个StopIteration的异常,来表示结束
关于yield from: yield from实现
迭代完一个可迭代对象 (直到可迭代对象抛出StopIteration). 觉得有点绕 ? 往下拉,看例子
yield from 补充例子 : yield from 生成器 协程
第一个生成器例子:
def f1():
print("f1")
yield 1 # 会暂停在这里
if __name__ == '__main__':
g = f1()
print("f1:" , f1) #还是一个函数
print("g:",g) #这个是生成器啊
value_from_yield = next(g) #来驱动一下
print(value_from_yield)
#问题是, 此时生成器结束了吗? 没呢!!! 只是暂停了 , 想继续怎么办? next !!!
try:
next(g) # 这时,yield 处返回,代码继续, 一旦生成器执行完会raise StopIteration
except StopIteration as e:
print(" g is done ")
下面开始把生成器和协程混合的讲:
协程由生成器进化来的,增强型的, 增强在哪里呢? 多了一个send,以及 yield不仅能产出值,还能接受值,
而生成器仅仅是一个生产者的角色:
def gen():
for i in range(10):
yield i
因此,下面代码中把协程 和 生成器混在一起说,他们的行为模式都是一样的;
生成器(协程)可以使用 next 来驱动代码继续 / send来交互.(驱动) .send 唯一比next多的一个功能就是发送一个值