Python中return和yield的区别

Python中return和yield的区别

return

在函数中,结束该函数的运行,return 后面的代码块不会执行,返回该函数的执行结果。

yield

带有yield的函数是一个迭代器,函数返回某个值时,会停留在某个位置,返回函数值后,会在前面停留的位置继续执行,直到程序结束

该对象可以迭代遍历和通过next()方法取出对象中的值。比较节约内存空间。保存的是生成数据的方式。可以达到随用随取的效果。

代码分析

看下面的程序:

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

扫描二维码关注公众号,回复: 12895048 查看本文章

4

解释代码运行顺序:

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是发送一个参数给res的,因为上面讲到,return的时候,并没有把4赋值给res,下次执行的时候只好继续执行赋值操作,只好赋值为None了,而如果用send的话,开始执行的时候,先接着上一次(return 4之后)执行,先把7赋值给了res,然后执行next的作用,遇见下一回的yield,return出结果后结束。

1.程序执行g.send(7),程序会从yield关键字那一行继续向下运行,send会把7这个值赋值给res变量

2.由于send方法中包含next()方法,所以程序会继续向下运行执行print方法,然后再次进入while循环

3.程序执行再次遇到yield关键字,yield会返回后面的值后,程序再次暂停,直到再次调用next方法或send方法。

这就结束了。这种方式比较节约内存空间。保存的是生成数据的方式。可以达到随用随取的效果。

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

总结

return对应的是普通函数,执行完毕以后控制权移交给父函数,函数的上下文就此销毁;

yeild对应的是一个具有迭代功能的生成器。无法用执行语句的方法调用它,必须用next()和send()函数进行调用。

next() 调用使生成器函数一直运行到下一条yield语句为止, 此时next()将返回值传递给yield,而且函数将暂时中止执行,再次调用next()时,函数将继续执行yield之后的语句。此过程将持续到函数返回为止。通常会使用一个for循环调用next()。

猜你喜欢

转载自blog.csdn.net/Kwoky/article/details/108721148