Python yield 使用方法浅析

发布时间 - 2026-01-11 01:12:23    点击率:

如何生成斐波那契數列

斐波那契(Fibonacci)數列是一个非常简单的递归数列,除第一个和第二个数外,任意一个数都可由前两个数相加得到。用计算机程序输出斐波那契數列的前 N 个数是一个非常简单的问题,许多初学者都可以轻易写出如下函数:

清单 1. 简单输出斐波那契數列前 N 个数

 def fab(max): 
  n, a, b = 0, 0, 1 
  while n < max: 
    print b 
    a, b = b, a + b 
    n = n + 1

执行 fab(5),我们可以得到如下输出:

 >>> fab(5)
 1
 1
 2
 3
 5

结果没有问题,但有经验的开发者会指出,直接在 fab 函数中用 print 打印数字会导致该函数可复用性较差,因为 fab 函数返回 None,其他函数无法获得该函数生成的数列。

要提高 fab 函数的可复用性,最好不要直接打印出数列,而是返回一个 List。以下是 fab 函数改写后的第二个版本:

清单 2. 输出斐波那契數列前 N 个数第二版

 def fab(max): 
  n, a, b = 0, 0, 1 
  L = [] 
  while n < max: 
    L.append(b) 
    a, b = b, a + b 
    n = n + 1 
  return L

可以使用如下方式打印出 fab 函数返回的 List:

 >>> for n in fab(5):
 ...     print n
 ...
 1
 1
 2
 3
 5

改写后的 fab 函数通过返回 List 能满足复用性的要求,但是更有经验的开发者会指出,该函数在运行中占用的内存会随着参数 max 的增大而增大,如果要控制内存占用,最好不要用 List来保存中间结果,而是通过 iterable 对象来迭代。例如,在 Python2.x 中,代码:

清单 3. 通过 iterable 对象来迭代

 for i in range(1000): pass会导致生成一个 1000 个元素的 List,而代码:

 for i in xrange(1000): pass则不会生成一个 1000 个元素的 List,而是在每次迭代中返回下一个数值,内存空间占用很小。因为 xrange 不返回 List,而是返回一个 iterable 对象。

利用 iterable 我们可以把 fab 函数改写为一个支持 iterable 的 class,以下是第三个版本的 Fab:

清单 4. 第三个版本

class Fab(object): 

  def __init__(self, max): 
    self.max = max 
    self.n, self.a, self.b = 0, 0, 1 

  def __iter__(self): 
    return self 

  def next(self): 
    if self.n < self.max: 
      r = self.b 
      self.a, self.b = self.b, self.a + self.b 
      self.n = self.n + 1 
      return r 
    raise StopIteration()

Fab 类通过 next() 不断返回数列的下一个数,内存占用始终为常数:

 >>> for n in Fab(5):
 ...     print n
 ...
 1
 1
 2
 3
 5

然而,使用 class 改写的这个版本,代码远远没有第一版的 fab 函数来得简洁。如果我们想要保持第一版 fab 函数的简洁性,同时又要获得 iterable 的效果,yield 就派上用场了:

清单 5. 使用 yield 的第四版

 def fab(max): 
  n, a, b = 0, 0, 1 
  while n < max: 
    yield b 
    # print b 
    a, b = b, a + b 
    n = n + 1 

'''

第四个版本的 fab 和第一版相比,仅仅把 print b 改为了 yield b,就在保持简洁性的同时获得了 iterable 的效果。

调用第四版的 fab 和第二版的 fab 完全一致:

 >>> for n in fab(5):
 ...     print n
 ...
 1
 1
 2
 3
 5

简单地讲,yield 的作用就是把一个函数变成一个 generator,带有 yield 的函数不再是一个普通函数,Python 解释器会将其视为一个 generator,调用 fab(5) 不会执行 fab 函数,而是返回一个 iterable 对象!在 for 循环执行时,每次循环都会执行 fab 函数内部的代码,执行到 yield b 时,fab 函数就返回一个迭代值,下次迭代时,代码从 yield b 的下一条语句继续执行,而函数的本地变量看起来和上次中断执行前是完全一样的,于是函数继续执行,直到再次遇到 yield。

也可以手动调用 fab(5) 的 next() 方法(因为 fab(5) 是一个 generator 对象,该对象具有 next() 方法),这样我们就可以更清楚地看到 fab 的执行流程:

清单 6. 执行流程

 >>> f = fab(5) 
 >>> f.next() 
 1 
 >>> f.next() 
 1 
 >>> f.next() 
 2 
 >>> f.next() 
 3 
 >>> f.next() 
 5 
 >>> f.next() 
 Traceback (most recent call last): 
 File "<stdin>", line 1, in <module> 
 StopIteration

当函数执行结束时,generator 自动抛出 StopIteration 异常,表示迭代完成。在 for 循环里,无需处理 StopIteration 异常,循环会正常结束。

我们可以得出以下结论:

一个带有 yield 的函数就是一个 generator,它和普通函数不同,生成一个 generator 看起来像函数调用,但不会执行任何函数代码,直到对其调用 next()(在 for 循环中会自动调用 next())才开始执行。虽然执行流程仍按函数的流程执行,但每执行到一个 yield 语句就会中断,并返回一个迭代值,下次执行时从 yield 的下一个语句继续执行。看起来就好像一个函数在正常执行的过程中被 yield 中断了数次,每次中断都会通过 yield 返回当前的迭代值。

yield 的好处是显而易见的,把一个函数改写为一个 generator 就获得了迭代能力,比起用类的实例保存状态来计算下一个 next() 的值,不仅代码简洁,而且执行流程异常清晰。

如何判断一个函数是否是一个特殊的 generator 函数?可以利用 isgeneratorfunction 判断:

清单 7. 使用 isgeneratorfunction 判断

 >>> from inspect import isgeneratorfunction 
 >>> isgeneratorfunction(fab) 
 True

要注意区分 fab 和 fab(5),fab 是一个 generator function,而 fab(5) 是调用 fab 返回的一个 generator,好比类的定义和类的实例的区别:

清单 8. 类的定义和类的实例

 >>> import types 
 >>> isinstance(fab, types.GeneratorType) 
 False 
 >>> isinstance(fab(5), types.GeneratorType) 
 True
fab 是无法迭代的,而 fab(5) 是可迭代的:
 >>> from collections import Iterable 
 >>> isinstance(fab, Iterable) 
 False 
 >>> isinstance(fab(5), Iterable) 
 True

每次调用 fab 函数都会生成一个新的 generator 实例,各实例互不影响:
 

>>> f1 = fab(3) 
 >>> f2 = fab(5) 
 >>> print 'f1:', f1.next() 
 f1: 1 
 >>> print 'f2:', f2.next() 
 f2: 1 
 >>> print 'f1:', f1.next() 
 f1: 1 
 >>> print 'f2:', f2.next() 
 f2: 1 
 >>> print 'f1:', f1.next() 
 f1: 2 
 >>> print 'f2:', f2.next() 
 f2: 2 
 >>> print 'f2:', f2.next() 
 f2: 3 
 >>> print 'f2:', f2.next() 
 f2: 5

return 的作用

在一个 generator function 中,如果没有 return,则默认执行至函数完毕,如果在执行过程中 return,则直接抛出 StopIteration 终止迭代。

另一个例子

另一个 yield 的例子来源于文件读取。如果直接对文件对象调用 read() 方法,会导致不可预测的内存占用。好的方法是利用固定长度的缓冲区来不断读取文件内容。通过 yield,我们不再需要编写读文件的迭代类,就可以轻松实现文件读取:

清单 9. 另一个 yield 的例子

 def read_file(fpath): 
  BLOCK_SIZE = 1024 
  with open(fpath, 'rb') as f: 
    while True: 
      block = f.read(BLOCK_SIZE) 
      if block: 
        yield block 
      else: 
        return

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。


# python  # yield  # 使用  # 用法  # 在Python中如何使用yield  # python如何正确使用yield  # 通过实例简单了解python yield使用方法  # python使用yield压平嵌套字典的超简单方法  # Python yield的使用详解  # 迭代  # 是一个  # 我们可以  # 一个函数  # 复用  # 第三个  # 递归  # 是一个非常  # 抛出  # 就可以  # 过程中  # 第四版  # 获得了  # 下次  # 就会  # 是在  # 就在  # 第一个  # 将其  # 对其 


相关栏目: 【 网站优化151355 】 【 网络推广146373 】 【 网络技术251813 】 【 AI营销90571


相关推荐: 香港服务器网站推广:SEO优化与外贸独立站搭建策略  node.js报错:Cannot find module &#39;ejs&#39;的解决办法  大连 网站制作,大连天途有线官网?  Laravel如何处理异常和错误?(Handler示例)  Laravel如何实现API版本控制_Laravel版本化API设计方案  制作ppt免费网站有哪些,有哪些比较好的ppt模板下载网站?  HTML5空格在Angular项目里怎么处理_Angular中空格的渲染问题【详解】  如何在建站主机中优化服务器配置?  如何彻底删除建站之星生成的Banner?  如何用5美元大硬盘VPS安全高效搭建个人网站?  Laravel怎么实现观察者模式Observer_Laravel模型事件监听与解耦开发【指南】  Python制作简易注册登录系统  Laravel怎么使用Session存储数据_Laravel会话管理与自定义驱动配置【详解】  大学网站设计制作软件有哪些,如何将网站制作成自己app?  如何在云虚拟主机上快速搭建个人网站?  如何将凡科建站内容保存为本地文件?  Win11任务栏卡死怎么办 Windows11任务栏无反应解决方法【教程】  软银砸40亿美元收购DigitalBridge 强化AI资料中心布局  无锡营销型网站制作公司,无锡网选车牌流程?  Zeus浏览器网页版官网入口 宙斯浏览器官网在线通道  米侠浏览器网页背景异常怎么办 米侠显示修复  网站制作壁纸教程视频,电脑壁纸网站?  Laravel如何实现多对多模型关联?(Eloquent教程)  郑州企业网站制作公司,郑州招聘网站有哪些?  如何在万网自助建站平台快速创建网站?  Laravel如何处理CORS跨域问题_Laravel项目CORS配置与解决方案  在Oracle关闭情况下如何修改spfile的参数  如何在IIS管理器中快速创建并配置网站?  Laravel如何为API编写文档_Laravel API文档生成与维护方法  如何用AWS免费套餐快速搭建高效网站?  如何快速搭建高效服务器建站系统?  Laravel如何理解并使用服务容器(Service Container)_Laravel依赖注入与容器绑定说明  如何获取免费开源的自助建站系统源码?  高配服务器限时抢购:企业级配置与回收服务一站式优惠方案  Laravel如何为API生成Swagger或OpenAPI文档  深圳网站制作培训,深圳哪些招聘网站比较好?  如何在自有机房高效搭建专业网站?  Python自动化办公教程_ExcelWordPDF批量处理案例  如何在阿里云域名上完成建站全流程?  JavaScript 输出显示内容(document.write、alert、innerHTML、console.log)  手机网站制作平台,手机靓号代理商怎么制作属于自己的手机靓号网站?  Laravel如何升级到最新的版本_Laravel版本升级流程与兼容性处理  如何快速搭建二级域名独立网站?  Thinkphp 中 distinct 的用法解析  phpredis提高消息队列的实时性方法(推荐)  高端云建站费用究竟需要多少预算?  Laravel如何处理表单验证?(Requests代码示例)  Laravel如何将应用部署到生产服务器_Laravel生产环境部署流程  Windows Hello人脸识别突然无法使用  悟空识字如何进行跟读录音_悟空识字开启麦克风权限与录音