python中生成器的使用

python中我们常听到迭代器和生成器,但是本人分开来介绍,只为告诉大家迭代器和生成器不是一个东西,在上篇文章中我们详细的介绍过迭代器和可迭代对象,本章重点介绍生成器。

一、生成器的应用场景

对于调用一个普通的Python函数,一般是从函数的第一行代码开始执行,结束于return语句、异常或者函数所有语句执行完毕。一旦函数将控制权交还给调用者,就意味着全部结束。函数中做的所有工作以及保存在局部变量中的数据都将丢失。再次调用这个函数时,一切都将从头创建。Python是通过生成器来实现类似于协同程序的概念:生成器可以暂时挂起函数,并保留函数的局部变量等数据,然后在再次调用它的时候,从上次暂停的位置继续执行下去。

二、一个最简单的生成器

  • 1、使用列表推导式的方式生成一个生成器

    from collections.abc import Iterator, Iterable
    
    if __name__ == "__main__":
        # 列表生成式
        lis = [x * x for x in range(10)]
        print(lis)
        print(isinstance(lis, Iterable))
        print(isinstance(lis, Iterator))
    
        # 生成器
        generator_ex = (x * x for x in range(10))
        print(generator_ex)
        print(isinstance(generator_ex, Iterable))
        print(isinstance(generator_ex, Iterator))
        print(next(generator_ex))
    复制代码
  • 2、从上面我们可以知道生成器是迭代器,具有迭代器的可迭代和**使用next()**函数的功能。记住:生成器是一种特殊的迭代器

三、为什么要使用生成器函数,而不是直接使用函数

  • 1、通过列表生成式可以直接创建一个列表,但是受到内存的限制,可能创建一个很大的空间,但是实际情况中用不到那么多(在C语言中所谓的堆区,没释放)
  • 2、生成器(generator)的定义就是根据列表元素可以动态的分配内存空间
  • 3、在实际开发过程中,如果我们要读取一个十几G大的文件,如果是直接使用文件打开的方式,其实底层的全部加载在内存中,这样造成计算机内存消耗,造成计算机卡死的局面,如果我们使用生成器,把大文件做成文档碎片的方式,每次从中间读取一点出来,然后再释放内存,这样就不会对计算机造成卡死的局面。

四、创建生成器函数

  • 1、将普通函数的return改为yield就是一个生成器函数

    def func():
        print('====>first')
        yield 1
        print('====>second')
        yield 2
        print('====>third')
        yield 3
        print('====>end')
    
    if __name__ == '__main__':
        g = func()
        print(g)
        # 生成器是一种特殊的迭代器,具有可迭代属性
        for item in g:
            print(item)
    复制代码

五、协同程序

协同程序(协程)一般来说是指这样的函数

  • 1、彼此间有不同的局部变量、指令指针,但是共享全局变量
  • 2、可以方便的挂起、恢复,并且有多个入口点和出口点
  • 3、多个协同程序间表现为协作运行,如A的运行过程中需要B的结果才能继续执行。

几个方法的介绍

  • 1、send(value):

    send()是除了next外另一个恢复生成器的方法,或者叫往生成器中传递参数的方法。是将yield语句变成了yield表达式,这意味着yield现在可以有一个值,而这个值就是在生成器的send方法被调用从而恢复执行时,调用send方法的参数。

    def test():
        i = 1
        while i < 5:
            temp = yield i ** 2
            print('temp', temp, '当前i=', i)
            i += 1
            
    if __name__ == "__main__":
        t = test()
        print(next(t))
        print(next(t))
        print(t.send('hello'))
        print(t.send('word'))
    复制代码
    # 运行结果
    1  							# 在main中打印的
    temp None 当前i= 1      # 在test函数中打印的
    4							# 在main中打印的
    temp hello 当前i= 2     # 在test函数中打印的
    9							# 在main中打印的
    temp word 当前i= 3      # 在test函数中打印的
    16							# 在main中打印的
    复制代码
    • 调用send传入非None值前,生成器必须处于挂起状态,否则将抛出异常
    • 第一次调用next的时候 yield 1**2,返回是1,当时temp没有传递值是None
    • 第二次调用next的时候 yield 2**2返回是4,当时temp没有传递值是None
    • 第三次调用send发送数据,发送了hello当时temp=word,打印出temp的值,yield 3**2返回是9
    • 第四次和第三次一样的
  • 2、close方法

    这个方法用于关闭生成器。对关闭的生成器后再次调用nextsend将抛出StopIteration异常

六、yield from的使用

yield fromPython3.3 后新加的语言结构。yield from的主要功能是打开双向通道,把最外层的调用方法与最内层的子生成器连接起来。这两者就可以进行发送值和返回值了,yeild from结构的本质是简化嵌套的生产器,不理解这个是什么意思的话,下面我将用几个例子来对其使用方法进行讲解

一般场景使用方式

def gene():
    for c in 'AB':
        yield c  # 遇到yeild程序返回循环,下次从yeild后面开始。
    for i in range(3):
        yield i

if __name__ == "__main__":
    print(list(gene()))  # list内部会预激生成器
复制代码

使用yield..from的时候

def gene1():
    yield from 'ab'
    yield from range(3)

if __name__ == "__main__":
    print(list(gene1()))   
复制代码

你可能感兴趣的:(python中生成器的使用)