yield关键字
在一个函数内实用yield关键字将函数变成生成器。
一个简单的例子：
# 一个迭代器函数
def fun(n):
for i in range(n):
yield i
print('--分割线--')
# 获取迭代器
a = fun(5)
# 使用next让迭代器继续执行
print(next(a))
print(next(a))
print(a.__next__())
print(a.__next__())解释：
- 定义一个生成器a，运行完第三行停止
- 第一个next函数执行，接收第一个i，此时i=0，输出0，第三行后的代码不执行
- 第二个next函数执行，从第四行执行代码，输出分割线后，进入下一个for循环，接收第二个i，此时i=1，输出1，第三行后的代码不执行
- 第三个next方法执行，从第四行执行代码，输出分割线后，进入下一个for循环，接收第二个i，此时i=2，输出2，第三行后的代码不执行
抽象解释： 将yield看成两个部分，一：return i，二：一个锁，每次执行next函数后，程序运行在一和二之间。下一次调用next函数先开第二部分的锁，输出分割线后进入一个新的循环，再return一个新的i，被新的锁锁住。（第一次相当于没有锁，直接return后再被锁）
tip：next函数和__next__()方法效果相同
关闭生成器：
# 关闭后无法用next进行迭代（用的少）
a.close()向生成器发送数据：
改一下代码：
# 一个迭代器函数
def fun(n):
for i in range(n):
k = yield i
if k:
print(k)
print('--分割线--')
# 获取迭代器
a = fun(5)
print(next(a))
# 发送一个字符
print(a.send('a'))
print(a.__next__())
print(a.__next__())可以看到a被输出了，不影响其他代码运行（send后同样进行一次迭代，结果和next相同）
传入一个异常：
# 此时生成器会抛出一个错误（可以用try语句接收）
a.throw(Exception)
# 一般这样接收
try:
yield 1
except Exception:
yield 'Exception'
finally:
print('finally')本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2023 年 05 月，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录