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

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

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


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显示处理传值给子生成器以及抛出异常给子生成器的意思了。

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

相关推荐

《Queendom》宣布冠军!女团MAMAMOO四人激动落泪

网易娱乐11月1日报道据台湾媒体报道,南韩女团竞争回归的生死斗《Queendom》昨(10/31)晚播出大决赛,并以直播方式进行,6组女团、女歌手皆演唱新歌,并加总前三轮的赛前赛、音源成绩与直播现场投...

正确复制、重写别人的代码,不算抄袭

我最近在一篇文章提到,工程师应该怎样避免使用大量的库、包以及其他依赖关系。我建议的另一种方案是,如果你没有达到重用第三方代码的阈值时,那么你就可以自己编写代码。在本文中,我将讨论一个在重用和从头开始编...

HTML DOM tr 对象_html event对象

tr对象tr对象代表了HTML表格的行。HTML文档中出现一个<tr>标签,就会创建一个tr对象。tr对象集合W3C:W3C标签。集合描述W3Ccells返回...

JS 打造动态表格_js如何动态改变表格内容

后台列表页最常见的需求:点击表头排序+一键全选。本文用原生js代码实现零依赖方案,涵盖DOM查询、排序算法、事件代理三大核心技能。效果速览一、核心思路事件入口:为每个<th>绑...

连肝7个晚上,总结了66条计算机网络的知识点

作者|哪吒来源|程序员小灰(ID:chengxuyuanxiaohui)计算机网络知识是面试常考的内容,在实际工作中经常涉及。最近,我总结了66条计算机网络相关的知识点。1、比较http0....

Vue 中 强制组件重新渲染的正确方法

作者:MichaelThiessen译者:前端小智来源:hackernoon有时候,依赖Vue响应方式来更新数据是不够的,相反,我们需要手动重新渲染组件来更新数据。或者,我们可能只想抛开当前的...

为什么100个前端只有1人能说清?浏览器重排/重绘深度解析

面试现场的"致命拷问""你的项目里做过哪些性能优化?能具体讲讲重排和重绘的区别吗?"作为面试官,我在秋招季连续面试过100多位前端候选人,这句提问几乎成了必考题。但令...

HTML DOM 介绍_dom4j html

HTMLDOM(文档对象模型)是一种基于文档的编程接口,它是HTML和XML文档的编程接口。它可以让开发人员通过JavaScript或其他脚本语言来访问和操作HTML和XML文档...

JavaScript 事件——“事件流和事件处理程序”的注意要点

事件流事件流描述的是从页面中接收事件的顺序。IE的事件流是事件冒泡流,而NetscapeCommunicator的事件流是事件捕获流。事件冒泡即事件开始时由最具体的元素接收,然后逐级向上传播到较为不...

探秘 Web 水印技术_水印制作网页

作者:fransli,腾讯PCG前端开发工程师Web水印技术在信息安全和版权保护等领域有着广泛的应用,对防止信息泄露或知识产品被侵犯有重要意义。水印根据可见性可分为可见水印和不可见水印(盲水印)...

国外顶流网红为流量拍摄性侵女学生?仅被封杀三月,回归仍爆火

曾经的油管之王,顶流网红DavidDobrik复出了。一切似乎都跟他因和成员灌酒性侵女学生被骂到退网之前一样:住在950万美元的豪宅,开着20万美元的阿斯顿马丁,每条视频都有数百万观看...人们仿佛...

JavaScript 内存泄漏排查方法_js内存泄漏及解决方法

一、概述本文主要介绍了如何通过Devtools的Memory内存工具排查JavaScript内存泄漏问题。先介绍了一些相关概念,说明了Memory内存工具的使用方式,然后介绍了堆快照的...

外贸独立站,网站优化的具体内容_外贸独立站,网站优化的具体内容有哪些

Wordpress网站优化,是通过优化代码、数据库、缓存、CSS/JS等内容,提升网站加载速度、交互性和稳定性。网站加载速度,是Google搜索引擎的第一权重,也是SEO优化的前提。1.优化渲染阻塞。...

这8个CSS工具可以提升编程速度_css用什么编译器

下面为大家推荐的这8个CSS工具,有提供函数的,有提供类的,有提取代码的,还有收集CSS的统计数据的……请花费两分钟的时间看完这篇文章,或许你会找到意外的惊喜,并且为你的编程之路打开了一扇新的大门。1...

vue的理解-vue源码 历史 简介 核心特性 和jquery区别 和 react对比

一、从历史说起Web是WorldWideWeb的简称,中文译为万维网我们可以将它规划成如下的几个时代来进行理解石器时代文明时代工业革命时代百花齐放时代石器时代石器时代指的就是我们的静态网页,可以欣...