生成器
- 生成器 generator
- 生成器指的是生成器对象,可以由生成器表达式得到,也可以使用yield关键字得到一个生成器函数,调用这个函数得到一个生成器对象
- 生成器对象,是一个可迭代对象,是一个迭代器
- 生成器对象,是延迟计算、惰性求值的
- 生成器函数
- 函数体中包含 yield语句的函数,就是生成器函数,调用后返回生成器对象
- 普通函数调用,函数会立即执行直到执行完毕,生成器函数调用,并不会立即执行函数体,而是需要使用next函数来驱动生成器函数执行后获得的生成器对象。
- 生成器表达式和生成器函数都可以得到生成器对象,只不过生成器函数可以写的更加复杂的逻辑
生成器的执行
- 在生成器函数中,可以多次yield,每执行一次yield后会暂停执行,把yield表达式的值返回
- 再次执行会执行到下一个yield语句又会暂停执行
- return语句依然可以终止函数运行,但return语句的返回值不能被获取到
- return会导致当前函数返回,无法继续执行,也无法继续获取下一个值,抛出StopIteration异常如果函数没有显式的return语句,如果生成器函数执行到结尾(相当于执行了return None),一样会抛出StopIteration异常
生成器函数
- 包含yield语句的生成器函数调用后,生成生成器对象的时候,生成器函数的函数体不会立即执行
- next(generator) 会从函数的当前位置向后执行到之后碰到的第一个yield语句,会弹出值,并暂停函数执
- 再次调用next函数,和上一条一样的处理过程
- 继续调用next函数,生成器函数如果结束执行了(显式或隐式调用了return语句),会抛出StopIteration异常
生成器应用
1.无限循环
def counter():
i = 0
while True:
i += 1
yield i
c = counter()
print(next(c))
print(next(c))
print(next(c))
2、计数器
def inc():
def counter():
i = 0
while True:
i += 1
yield i
c = counter()
def inner():
return next(c)
return inner # return lambda : next(c)
foo = inc()
print(foo()) # 打印什么?
print(foo()) # 打印什么?
print(foo()
# 代码中的inner函数可以由lambda表达式替代
3.斐波那契数列
def fib():
x = 0
y = 1
while True:
yield y
x, y = y,x + y
foo = fib()
for i in range(10): # 打印前10行
print(next(foo))
4 .生成器交互
- python提供了一个和生成器对象交互的方法send,该方法可以和生成器沟通
def inc():
def counter():
i = 0
while True:
i += 1
response = yield i
if response is not None:
i = response
c = counter()
return lambda x=False: next(c) if not x else c.send(0)
foo = inc()
print(foo())
print(foo())
print(foo())
print(foo(True))
print(foo())
print(foo())
print(foo())
打印结果 :
1
2
3
1
2
3
4
- 调用send方法,就可以把send的实参传给yield语句做结果,这个结果可以在等式右边被赋值给其它变量
- send和next一样可以推动生成器启动并执行
5.协程 Coroutine
- 生成器的高级用法
- 它比进程、线程轻量级,是在用户空间调度函数的一种实现
- Python3 asyncio就是协程实现,已经加入到标准库
- Python3.5 使用async、await关键字直接原生支持
- 协程协程调度器实现思路有
- 2个生成器A、B
- next(A)后,A执行到了yield语句暂停,然后去执行next(B),B执行到yield语句也暂停,然后再次调用
- next(A),再调用next(B)在,周而复始,就实现了调度的效果
- 可以引入调度的策略来实现切换的方式
- 协程是一种非抢占式调度
- yield from语法
- yield from iterable等价于for item in iterable: yield item。
def inc():
for x in range(1000):
yield x
# 使用yield from 简化
def inc():
yield from range(1000)
foo = inc()
print(next(foo))
print(next(foo))
print(next(foo))
本质上yield from的意思就是,从from后面的可迭代对象中拿元素一个个yield出去。
更多推荐
函数生成器
发布评论