百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术文章 > 正文

python中你必须要了解的yield和yield from

itomcoil 2025-02-17 12:29 21 浏览


yield和yield from用法

迭代器(iterator)

讲到迭代器,就需要区别几个概念:iterable, iterator, itertion, 看着都差不多,其实不然。下面区分一下。

  • iterable这个是可迭代对象,属于python的名词,范围也很广,可重复迭代,满足如下其中之一的都是iterable可以for循环:for i in iterable可以按index索引的对象,也就是定义了__getitem__方法,比如list, str定义了__iter__方法,可以随意返回可以调用iter(obj)的对象,并且返回一个iterator
  • iterator迭代器对象,也属于python的名词,只能迭代一次。需要满足如下的迭代器协议定义了__iter__方法,但是必须返回自身定义了next方法,在python3.x是__next__。用来返回下一个值,并且当没有数据了,抛出StopIteration可以保持当前的状态
  • itertion就是迭代,一个接一个(one after another),是一个通用的概念,比如一个循环遍历某个数组。

凡是可作用于for循环的对象都是Iterable类型凡是可作用于next()函数的对象都是Iterator类型,它们表示一个惰性计算的序列集合数据类型如list、dict、str等是Iterable但不是Iterator,不过可以通过iter()函数获得一个Iterator对象

yield

generator的函数,在每次调用next()的时候执行,遇到yield语句返回,在此执行时从上次返回的yield语句处继续执行。

  • 创建生成器的方法yield关键字生成器表达式 (i for i in range(5))自定义一个有next方法和iter方法的类
  • next方法语法: next(iterator[, default])作用:next()返回迭代器的下一个项目。iterator-可迭代对象. default-可选,用于设置在没有下一个元素时返回该默认值,如果不设置,又没有下一个元素则会触发Stopiteration异常。因为generator保存的是算法,每次调用next(),就计算出下一个元素的值,直到计算到最后一个元素,没有更多的元素时,抛出Stopiteration的错误。返回值:生成器中也可以包含return语句,但是不能出现在yield表达式中,当执行到return语句时,如果有finally语块则执行,之后会抛出StopIteration异常。不再继续其他的迭代生成。
  • iter方法语法:iter(object[, sentinel])作用:返回一个iterator对象
  • send方法语法:send(value)作用:send方法会首先把上一次挂起的yield语句的返回值通过参数设定,从而实现与生成器方法的交互。但是需要注意在一个生成器对象没有执行next方法之前,由于没有yield语句被挂起,所以执行send方法会报错,除非执行sene(None)
  • throw方法作用
    它的实现手段是通过向生成器对象在上次被挂起处,跑出一个异常。之后会继续执行生成器对象中后面的语句,直到遇到下一个yield语句返回。如果在生成器对象方法执行完毕后,依然没有遇到yield语句,跑出StopIteration异常。
  • close方法close方法会在生成器对象方法的挂起处跑出一个GeneratorExit异常。GeneratorExit异常产生后,系统会继续把生成器对象方法后续的代码执行完毕。

需要注意的是GeneratorExit异常的产生意味着生成器对象的生命周期已经结束。因此,一旦产生了GeneratorExit异常,生成器方法后续执行的语句中,不能再有yield语句,否则会产生RuntimeError。

yield from

  • 出现原因生成器能够很容易分为多个拥有send和throw方法的子生成器,像一个大函数可以分为多个子函数一样简单。Python的生成器是协成coroutime的一种形式,但它的局限性在于只能向它的直接调用者yield值。这意味着那些包含yield的代码不能像其他代码那样被分离出来放到一个单独的函数中。
  • 对于简单的迭代器,yield from iterable本质上等于for item in iterable: yield item的缩写版,如下所示:
def g(x):
    yield from range(x , 0, -1)
    yield from range(x)

print(list(g(5)))
  • 然而,不同于普通的循环,yield from允许子生成器直接从调用者接收其发送的信息或者跑出调用时遇到的异常,并且返回给委派生成器一个值,如下所示:
def accumulate():  #子生成器
    tally = 0
    while True:
        next = yield
        if next is None:
            return tally
        tally = tally + next
   
def gather_tallies(tallies):
    while True:
        tally = yield from accumulate()
        tallies.append(tally)

tallies = []
acc = gather_tallies(tallies)
next(acc) # 激活生成器
for i in range(4):
    acc.send(i)
acc.send(None) # 结束
for i in range(5):
    acc.send(i)
acc.send(None)
print(tallies)
  • 利用yield from从生成器读取数据
def reader():
    # 模拟从文件读取数据的生成器
    for i in range(4):
        yield '***{}***'.format(i)

def reader_wrapper(g):
    # 循环迭代从reader产生的数据
    for v in g:
        yield v

wrap = reader_wrapper(reader())
for i in wrap:
    print(i)

# 输出结果:
***1***
***2***
***3***
***4***

我们可以用yield from语句替代reader_wrapper(g)函数中的循环,如下:def reader_wrapper(g):yield from g效果是一样的

  • 利用yield from语句向生成器(协程)传送数据
def writer():
    # 读取send传进的数据,并模拟写进套接字或文件
    while True:
        w = (yield) #w接收send传进的数据
        print('>>', w)

def writer_wrapper(coro1):
    coro1.send(None) #生成器准备好接收数据
    while True:
        try:
            x = (yield) # x接收send传进的数据
            coro1.send(x) #然后将x在send给writer子生成器
        except StopIteration: #处理子生成器返回的异常
            pass

# 包装器也是个生成器,上面所有复杂的写法也可以用yield from替换
def writer_wrapper(coro2):
    yield from coro2
  • 利用yield from向生成器传送数据--处理异常
class SpamException(Exception):
    pass

def writer():
    while True:
        try:
            w = (yield)
        except SpamException:
            print('***')
        else:
            print('>>', w)

def writer_wrapper(coro1):
    # 手工处理异常被抛给子生成器
    coro1.send(None) # 生成器准备好接收数据
    while True:
        try:
            try:
                x = (yield)
            except Exception as e: #捕获异常
                coro1.throw(e)
            else:
                coro1.send(x)
        except StopIteration:
            pass

w = writer()
wrap = writer_wrapper(w)
wrap.send(None) # "prime" the coroutine
for i in [0, 1, 2, 'spam', 4]:
    if i == 'spam':
        wrap.throw(SpamException)
    else:
        wrap.send(i)

# 同样的可以把writer_wrapper方法协成:
def writer_wrapper(coro):
    yield from coro

看到这里大概能理解yield from显示处理传值给子生成器以及抛出异常给子生成器的意思了。

总之,这是一个魔法语句,它也是协成的重要组成部分,至于协成,还需要继续学习。

相关推荐

selenium(WEB自动化工具)

定义解释Selenium是一个用于Web应用程序测试的工具。Selenium测试直接运行在浏览器中,就像真正的用户在操作一样。支持的浏览器包括IE(7,8,9,10,11),MozillaF...

开发利器丨如何使用ELK设计微服务中的日志收集方案?

【摘要】微服务各个组件的相关实践会涉及到工具,本文将会介绍微服务日常开发的一些利器,这些工具帮助我们构建更加健壮的微服务系统,并帮助排查解决微服务系统中的问题与性能瓶颈等。我们将重点介绍微服务架构中...

高并发系统设计:应对每秒数万QPS的架构策略

当面试官问及"如何应对每秒几万QPS(QueriesPerSecond)"时,大概率是想知道你对高并发系统设计的理解有多少。本文将深入探讨从基础设施到应用层面的解决方案。01、理解...

2025 年每个 JavaScript 开发者都应该了解的功能

大家好,很高兴又见面了,我是"高级前端进阶",由我带着大家一起关注前端前沿、深入前端底层技术,大家一起进步,也欢迎大家关注、点赞、收藏、转发。1.Iteratorhelpers开发者...

JavaScript Array 对象

Array对象Array对象用于在变量中存储多个值:varcars=["Saab","Volvo","BMW"];第一个数组元素的索引值为0,第二个索引值为1,以此类推。更多有...

Gemini 2.5编程全球霸榜,谷歌重回AI王座,神秘模型曝光,奥特曼迎战

刚刚,Gemini2.5Pro编程登顶,6美元性价比碾压Claude3.7Sonnet。不仅如此,谷歌还暗藏着更强的编程模型Dragontail,这次是要彻底翻盘了。谷歌,彻底打了一场漂亮的翻...

动力节点最新JavaScript教程(高级篇),深入学习JavaScript

JavaScript是一种运行在浏览器中的解释型编程语言,它的解释器被称为JavaScript引擎,是浏览器的一部分,JavaScript广泛用于浏览器客户端编程,通常JavaScript脚本是通过嵌...

一文看懂Kiro,其 Spec工作流秒杀Cursor,可移植至Claude Code

当Cursor的“即兴编程”开始拖累项目质量,AWS新晋IDEKiro以Spec工作流打出“先规范后编码”的系统工程思维:需求-设计-任务三件套一次生成,文档与代码同步落地,复杂项目不...

「晚安·好梦」努力只能及格,拼命才能优秀

欢迎光临,浏览之前点击上面的音乐放松一下心情吧!喜欢的话给小编一个关注呀!Effortscanonlypass,anddesperatelycanbeexcellent.努力只能及格...

JavaScript 中 some 与 every 方法的区别是什么?

大家好,很高兴又见面了,我是姜茶的编程笔记,我们一起学习前端相关领域技术,共同进步,也欢迎大家关注、点赞、收藏、转发,您的支持是我不断创作的动力在JavaScript中,Array.protot...

10个高效的Python爬虫框架,你用过几个?

小型爬虫需求,requests库+bs4库就能解决;大型爬虫数据,尤其涉及异步抓取、内容管理及后续扩展等功能时,就需要用到爬虫框架了。下面介绍了10个爬虫框架,大家可以学习使用!1.Scrapysc...

12个高效的Python爬虫框架,你用过几个?

实现爬虫技术的编程环境有很多种,Java、Python、C++等都可以用来爬虫。但很多人选择Python来写爬虫,为什么呢?因为Python确实很适合做爬虫,丰富的第三方库十分强大,简单几行代码便可实...

pip3 install pyspider报错问题解决

运行如下命令报错:>>>pip3installpyspider观察上面的报错问题,需要安装pycurl。是到这个网址:http://www.lfd.uci.edu/~gohlke...

PySpider框架的使用

PysiderPysider是一个国人用Python编写的、带有强大的WebUI的网络爬虫系统,它支持多种数据库、任务监控、项目管理、结果查看、URL去重等强大的功能。安装pip3inst...

「机器学习」神经网络的激活函数、并通过python实现激活函数

神经网络的激活函数、并通过python实现whatis激活函数感知机的网络结构如下:左图中,偏置b没有被画出来,如果要表示出b,可以像右图那样做。用数学式来表示感知机:上面这个数学式子可以被改写:...