前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >如何理解python中的yield,看完还是不懂?我手把手教你

如何理解python中的yield,看完还是不懂?我手把手教你

作者头像
胡八万
发布2022-06-08 15:42:55
5230
发布2022-06-08 15:42:55
举报
文章被收录于专栏:软件测试技术软件测试技术

关于yield 看了忘,忘了看,零零散散的总是理解不透彻。今天彻底记录下,带大家一探 yield到底是什么?

关于生成器概念的解释,摘自菜鸟教程解释:

在 Python 中,使用了 yield 的函数被称为生成器(generator)。跟普通函数不同的是,生成器是一个返回迭代器的函数,只能用于迭代操作,更简单点理解生成器就是一个迭代器。在调用生成器运行的过程中,每次遇到 yield 时函数会暂停并保存当前所有的运行信息,返回 yield 的值, 并在下一次执行 next() 方法时从当前位置继续运行。调用一个生成器函数,返回的是一个迭代器对象。

字都认识,但是连到一起就不认识了 : s。网上例子很多,所有例子一开始就是这样的代码:

代码语言:javascript
复制
# !!!这种代码 建议别看,因为以我亲身经历,初次理解生成器,看这样的代码根本看不懂的~

def evenNumber(max):
    n = 0
    while n < max:
        yield n
        n += 2
        
for i in evenNumber(10):
    print(i)

一会儿我把上述代码重新写下。

在此之前,大家还是需要掌握什么是generator(生成器)。

最简单的创建一个生成器的方式,只要把一个列表生成式的 [] 改成 () ,就创建了一个generator:

代码语言:javascript
复制
List  = [i*2 for i in range(10)]
print(List)
#[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

g = (x*2 for x in range(10))
print(g)
#<generator object <genexpr> at 0x000001CBA42F3C10>

生成器(generator)能够迭代的关键是它有一个next()方法,工作原理就是通过重复调用next()方法,直到捕获一个异常。试一下:

代码语言:javascript
复制
g = (x*2 for x in range(10))

print(g.__next__())
# 0
print(g.__next__())
# 2
print(g.__next__())
# 4
#... ...

第二种方法就是, 如果一个函数中包含yield关键字,那么这个函数就不再是一个普通函数,而是一个generator。调用函数就是创建了一个生成器(generator)对象。

着重讲一下第二种方法,先看代码,为了便于理解,我不写循环,写一个生成器函数:

代码语言:javascript
复制
def gen_example():
    print ('第1次执行啦~,还没到第一个yield!')
    yield '我是第1个遇见的yield,你遇到我就要返回'
    
    print ('第2次执行啦~,还没到第二个yield')
    yield '我是第2个遇见的yield,你遇到我就要返回'
    
    print ('第3次执行啦,我运行完 函数就执行完毕啦~')


for i in gen_example():
    print(i)
    print("--------分割线--------")


执行结果:

代码语言:javascript
复制
第1次执行啦~,还没到第一个yield!
我是第1个遇见的yield,你遇到我就要返回
--------分割线--------
第2次执行啦~,还没到第二个yield
我是第2个遇见的yield,你遇到我就要返回
--------分割线--------
第3次执行啦,我运行完 函数就执行完毕啦~

过程详解:

第一次for 循环执行到yield结束 ,只执行了这两句代码:

第二次循环 是从上一次的yield结束地方开始执行, 到下一个yield结束(一定要多读几遍理解哦):

第三次循环是,从第二次遇见的yield结束地方开始执行,一直到下一个yield结束(没有yield,自动执行结束)

通过上述规律我们不难发现,yield相当于 return 返回一个值,并且记住这个返回的位置,下次迭代时,代码从yield的下一条语句开始执行!

为了验证这个规律 我把本文开头的代码 重新写下,用更容易懂的方式:

代码语言:javascript
复制
def evenNumber(max):
    n = 0
    while n < max:
        yield n
        print("第二次是从这里开始的")
        n += 2
        print(f"n在第二次是{n}")


for i in evenNumber(2):

    print(i)
    print("--------分割线--------")

执行结果:

代码语言:javascript
复制
0
--------分割线--------
第二次是从这里开始的
n在第二次是2

输出:0, 是怎么得到的呢?函数第一次执行,遇到yield 就返回,所以此时就打印0。

第二次是从这里开始的,n在第二次是2这又是怎么得到的呢?执行yield后面的这部分,然后又进入循环,while循环条件都不满足,执行结束。实际第二次没有返回值。如果你细心点就会发现,第二次输出都没有分割线的内容:

既然 第二次打印都终止了,为什么没报错?好问题!因为调用next()来执行生成器则会报错,如果使用for循环遍历,for循环会自动捕获该异常,直接停止遍历。

代码语言:javascript
复制
def evenNumber(max):
    n = 0
    while n < max:
        yield n
        print("第二次是从这里开始的")
        n += 2
        print(f"n在第二次是{n}")

g = evenNumber(2)
print(g.__next__())
print(g.__next__())

执行结果,调用next方法就会报错:

代码语言:javascript
复制
0
第二次是从这里开始的
n在第二次是2
Traceback (most recent call last):
  File "test.py", line 17, in <module>
    print(g.__next__())
StopIteration

***Repl Closed***

所以,就印证了我们的结论~!现在来看一些简单的生成器函数的例子,是不是就一下懂了。

如果看到这里还是没懂 ,留言吧,我会好好反思我自;

留个小作业,大家可以试试分析下斐波那契数列的过程。

代码语言:javascript
复制
def fab(max): 
    n, a, b = 0, 0, 1 
    while n < max: 
        yield b     
        a, b = b, a + b 
        n = n + 1
 
for n in fab(5): 
    print n

其它补充,如何判断是不是 生成器函数?

使用 isgeneratorfunction 判断函数是否是 generator 函数

代码语言:javascript
复制
from inspect import isgeneratorfunction 
print(isgeneratorfunction(fab) )
#True

其它补充,为什么生成器函数中的return 不能返回值?

先看代码:

代码语言:javascript
复制
def evenNumber(max):
    n = 0
    while n < max:
        yield n
        print("第二次是从这里开始的")
        n += 2
        print(f"n在第二次是{n}")
        return 10


for i in evenNumber(10):

    print(i)
    print("--------分割线--------")

执行结果:

代码语言:javascript
复制
0
--------分割线--------
第二次是从这里开始的
n在第二次是2

***Repl Closed***

为什么 我的return 的值 没有在最后一次打印出来呢?在官方文档中是这样解释的:![在这里插入图片描述](https://img-blog.csdnimg.cn/c53ff0829a694466984afc0280d87052.png

在这里插入图片描述

python生成器函数中return 的作用就是终止生成器。表示生成器运行完成了,可以结束了。然后生成器会抛出一个StopIteration的异常。而for循环能够检测到这个异常,于是结束循环,也不报错。但是__next__()就会报错哦~

好了 ,结束。

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2022-05-17,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 软件测试技术 微信公众号,前往查看

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

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 其它补充,如何判断是不是 生成器函数?
  • 其它补充,为什么生成器函数中的return 不能返回值?
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档