Python中的yield到底是个什么鬼?

开发 后端
相信你已经不止一次在函数中看到关键词yield,它起着什么作用?返回什么?和return又有着什么区别呢?这篇文章将会揭开yield的神秘面纱,并给出最浅显易懂的例子。

相信你已经不止一次在函数中看到关键词yield,它起着什么作用?返回什么?和return又有着什么区别呢?这篇文章将会揭开yield的神秘面纱,并给出最浅显易懂的例子。

yield关键字做了什么?

如果不太好理解yield,可以先把yield当作return的同胞兄弟来看,他们都在函数中使用,并履行着返回某种结果的职责。

这两者的区别是:

有return的函数直接返回所有结果,程序终止不再运行,并销毁局部变量;

而有yield的函数则返回一个可迭代的 generator(生成器)对象,你可以使用for循环或者调用next()方法遍历生成器对象来提取结果。

什么是生成器呢?在 Python 中,使用了yield的函数被称为生成器。有点套娃的感觉,但事实就是这样,调用一个yield函数,就会返回一个生成器对象。

在调用生成器函数的过程中,每次遇到 yield 时函数会暂停并保存当前所有的运行信息(保留局部变量),返回yield的值, 并在下一次执行next()方法时从当前位置继续运行,直到生成器被全部遍历完。

先看一个简单的例子:

上面函数返回一个生成器,现在对该生成器进行遍历:

或者调用next()方法:

可以看到,simple_generator函数返回一个生成器,调用next()方法后,函数开始运行,遇到第一个yield关键字,返回生成的值(1),程序暂停;

第二次调用next()方法,代码从上次暂停的位置开始执行,并遇到了第二个yield关键字,再返回生成的值(2),程序暂停;

第三次调用也是如此,返回生成的值(3),生成器耗尽,程序终止;

到这里你可能就明白yield和return的关系和区别了,带yield的函数是一个生成器,这个生成器有一个方法就是next,next就相当于“下一步”生成哪个数,这一次的next开始的地方是接着上一次的next停止的地方执行的。

所以调用next的时候,生成器并不会从函数的开始执行,只是接着上一步停止的地方开始,然后遇到yield后,return出要生成的数,此步就结束。

有个经典的例子就是使用yield生成斐波那契数列:

  1. def fab(max):  
  2.     n, a, b = 0, 0, 1  
  3.     while n < max:  
  4.         yield b      # 使用 yield 
  5.   
  6.         a, bb = b, a + b  
  7.         nn = n + 1 
  8.   
  9. for n in fab(5):  
  10.     print n 

生成器有哪些作用?

如果想具体化数据的形式,通常会将数据存储在一个列表中。但这样做,列表的内容将占用有形内存。列表越大,占用的内存资源就越多。

但是,如果数据集有某种逻辑,就不必存储在一个列表中,只需编写一个生成器,它将在需要时生成这些值,基本不占用内存。

 

责任编辑:赵宁宁 来源: Python大数据分析
相关推荐

2022-02-16 20:04:08

容器KubernetesShim

2021-03-06 09:18:51

JS闭包函数

2020-09-27 06:53:57

MavenCDNwrapper

2019-10-30 10:13:15

区块链技术支付宝

2024-09-11 13:58:18

2024-07-12 15:08:23

Python@wraps函数

2022-02-22 13:20:57

RSA算法加密

2021-01-29 12:24:22

电脑电子计算机

2012-02-13 15:50:59

2022-04-10 19:26:07

TypeScript类型语法

2021-01-08 09:48:18

Pythonname变量

2022-01-10 11:16:40

漏洞 Log4j2Jndi

2021-02-18 07:43:25

TCP协议网络

2021-03-22 07:45:05

Sentinel微服务开源的项目

2024-01-16 23:30:46

​fractionsPython分数

2018-09-06 11:20:24

CDNDDoS网站

2024-08-26 14:23:56

2015-05-21 15:45:13

2016-10-21 09:58:19

WindowsKMSOEM系统

2024-07-03 12:04:42

C++this​
点赞
收藏

51CTO技术栈公众号