yield关键字

TomatoCool

发布于 2023-07-30 09:43:52

18600

代码可运行

文章被收录于专栏：TomatoCoolTomatoCool

运行总次数：0

代码可运行

在一个函数内实用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 删除

exception

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

登录后参与评论

暂无评论

yield关键字

yield关键字

一个简单的例子：

关闭生成器：

向生成器发送数据：

传入一个异常：

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐