python中yield的用法笔记

刚刚在看一个程序的时候,突然发现yield这个关键字,不是很明白,查阅了下相关资料,现留下笔记以便日后复看。

首先,在对yield还没有初步认识时,可以先把yield看做“return”,这个是直观,它首先是个return,普通的return是什么意思,就是在程序中返回某个值,返回之后程序就不再往下运行了。看做return之后再把它看做一个是生成器(generator)的一部分(带yield的函数才是真正的迭代器。下面我们从程序上来理解yield:

def foo():
    print("starting...")
    while True:
        res = yield 4
        print("res:",res)
g = foo()
print(next(g))
print("*"*20)
print(next(g))

输出为:

starting...
4
********************
res: None
4

按照代码的运行顺序,直接运行Debug更直观:

  1. 程序开始执行以后,因为foo函数中有yield关键字,所以foo函数并不会真的执行,而是先得到一个生成器g(相当于一个对象);
  2. 直到调用next方法,foo函数正式开始执行,先执行foo函数中的print方法,然后进入while循环;
  3. 程序遇到yield关键字,然后把yield想成return,return一个4之后,程序停止,并没有执行赋值给res操作,此时next(g)语句执行完成,所以输出的前两行(第一个是while上面的print的结果,第二个是return出的结果)是执行print(next(g))的结果;
  4. 程序执行print("*"*20),输出20个*;
  5. 又开始执行下面的print(next(g)),这个时候和上面那个差不多,不过不同的是,这个时候是从刚才那个next程序停止的地方开始执行的,也就是要执行res的赋值操作,这时候要注意,这个时候赋值操作的右边是没有值的(因为刚才那个是return出去了,并没有给赋值操作的左边传参数),所以这个时候res赋值是None,所以接着下面的输出就是res:None;
  6. 程序会继续在while里执行,又一次碰到yield,这个时候同样return 出4,然后程序停止,print函数输出的4就是这次return出的4。

我们可以从这里看出yield和return的关系和区别,带yield的函数是一个生成器,而不是一个函数了,这个生成器有一个函数就是next函数,next就相当于“下一步”生成哪个数,这一次的next开始的地方是接着上一次的next停止的地方执行的,所以调用next的时候,生成器并不会从foo函数的开始执行,只是接着上一步停止的地方开始,然后遇到yield后,return出要生成的数,此步就结束。

同时还有另一个生成器send函数:

def foo():
    print("starting...")
    while True:
        res = yield 4
        print("res:",res)
g = foo()
print(next(g))
print("*"*20)
print(g.send(7))

只是将上面的代码最后一行换了,我们来看下输出结果:

starting...
4
********************
res: 7
4

先大致说一下send函数的概念:上面我们提到,yield函数是没有给左边res赋值的,所以res的值是None,然而这里变成了7,到底为什么,这是因为,send是发送一个参数给res的,因为上面讲到,return的时候,并没有把4赋值给res,下次执行的时候只好继续执行赋值操作,只好赋值为None了,而如果用send的话,开始执行的时候,先接着上一次(return 4之后)执行,先把7赋值给了res,然后执行next的作用,遇见下一回的yield,return出结果后结束。

如果跑Debug,我们会发现,从上面第五步开始:

  1. 程序执行g.send(7),程序会从yield关键字那一行继续向下运行,send会把7这个值赋值给res变量;
  2. 由于send方法中包含next()方法,所以程序会继续向下运行执行print方法,然后再次进入while循环;
  3. 程序执行再次遇到yield关键字,yield会返回后面的值后,程序再次暂停,直到再次调用next方法或send方法。

提一下为什么要用到这个生成器,因为如果用list的话,会占用很大的空间,比如说取0,1,2,3·······999,1000;如果这样写:

for n in range(1000):
    a=nfor n in range(1000):
    a=n

此时的range(1000)就默认生成一个含有1000个数的list,所以很占空间。
这时就完全可以用刚才的yield组合成生成器进行实现:

def foo(num):
    print("starting...")
    while num<10:
        num=num+1
        yield num
for n in foo(0):
    print(n)

输出为:

starting...
1
2
3
4
5
6
7
8
9
10

若有那里不对,请帮忙指出,谢谢。

展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 深蓝海洋 设计师: CSDN官方博客
应支付0元
点击重新获取
扫码支付

支付成功即可阅读