python生成器的传参数与返回值

一.生成器简介

在python中,带yield的方法不再是普通方法,而是生成器,它的执行顺序不同与普通方法.

普通方法的执行是从头到尾,最后return返回,或者没有返回值

生成器是到yield就返回yield之后的值,然后阻塞,等待next()/send()继续调起生成器

二.next()/send()的异同:

next()/send()都可以调起生成器

next()只能调起生成器返回值,send()还可以对生成器进行传参数,与正常的理解不同,yield表达式左侧的值是send()方法传进来的参数

还有一个有意思的点,第一次send()的时候要求只能send(None)

三.生成器的执行顺序

在yield这一行代码的执行逻辑是这样的,等号右边的yield r先执行,将r返回,阻塞...

等到下次生成器再被调起的时候,yield把传入的参数给n,再继续执行下面的代码,知道yield继续返回...

def consumer(): 
    r = '' 
    while True: 
        n = yield r 
        if not n: 
            return 
        print('[CONSUMER] Consuming %s...' % n) 
        r = '200 OK' 
 
def produce(c): 
    c.send(None) 
    n = 0 
    while n < 5: 
        n = n + 1 
        print('[PRODUCER] Producing %s...' % n) 
        r = c.send(n) 
        print('[PRODUCER] Consumer return: %s' % r) 
    c.close() 
                                                                                                                       

In [31]: c=consumer()                                                                                                           

In [32]: c.send(None)                                                                                                           
Out[32]: ''

In [33]: c.send(1)                                                                                                              
[CONSUMER] Consuming 1...
Out[33]: '200 OK'

In [34]: c.send(2)                                                                                                              
[CONSUMER] Consuming 2...
Out[34]: '200 OK'