在一个函数内实用yield关键字将函数变成生成器。
一个简单的例子:
代码语言:javascript复制# 一个迭代器函数
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__()方法效果相同
关闭生成器:
代码语言:javascript复制# 关闭后无法用next进行迭代(用的少)
a.close()
向生成器发送数据:
改一下代码:
代码语言:javascript复制# 一个迭代器函数
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相同)
传入一个异常:
代码语言:javascript复制# 此时生成器会抛出一个错误(可以用try语句接收)
a.throw(Exception)
# 一般这样接收
try:
yield 1
except Exception:
yield 'Exception'
finally:
print('finally')