yield关键字
*在一个函数内实用yield关键字将函数变成生成器* ## 一个简单的例子 ```python # 一个迭代器函数 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__()) ``` ### 解释 1. 定义一个生成器a,运行完第三行停止 2. 第一个next函数执行,接收第一个i,此时i=0,输出0,第三行后的代码不执行 3. 第二个next函数执行,从第四行执行代码,输出分割线后,进入下一个for循环,接收第二个i,此时i=1,输出1,第三行后的代码不执行 4. 第三个next方法执行,从第四行执行代码,输出分割线后,进入下一个for循环,接收第二个i,此时i=2,输出2,第三行后的代码不执行 ### 抽象解释 将yield看成两个部分,一:return i,二:一个锁,每次执行next函数后,程序运行在一和二之间。下一次调用next函数先开第二部分的锁,输出分割线后进入一个新的循环,再return一个新的i,被新的锁锁住。(第一次相当于没有锁,直接return后再被锁) > tip:next函数和__next__()方法效果相同 ## 关闭生成器 ```python # 关闭后无法用next进行迭代(用的少) a.close() ``` ## 向生成器发送数据 改一下代码: ```python # 一个迭代器函数 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相同) ## 传入一个异常 ```python # 此时生成器会抛出一个错误(可以用try语句接收) a.throw(Exception) # 一般这样接收 try: yield 1 except Exception: yield 'Exception' finally: print('finally') ```
创建时间:2023-05-19
|
最后修改:2023-12-27
|
©允许规范转载
酷酷番茄
首页
文章
友链