python-迭代器与生成器

    科技2024-12-15  13

    目录

    迭代器

    迭代器的创建与使用

    生成器(generator)

    yield

    推导式变生成器


    迭代器

    迭代器在迭代元素的时候可以记住遍历的位置,迭代器只能前进不能后退。

    迭代器的创建与使用

    创建迭代器:iter() 括号中的参数为可迭代对象

    迭代器结果输出:next() 括号中的参数为创建出来的迭代器

    迭代器是一个类,迭代器类中定义有__iter__() 和__next__()两个方法。__iter__()方法用来返回一个迭代器对象,这个迭代器对象实现了__next__()方法并通过 StopIteration 异常标识迭代的完成。__next__() 方法会返回下一个迭代器对象。我们可以自己创建迭代器类来使用。

    使用迭代器时,如果迭代器中数据迭代完之后继续迭代就会报错。因此在使用next()迭代时,一定要做异常的处理。

    import sys # 引入sys模块 while True: try: print(next(迭代器)) except StopIteration: sys.exit()

    生成器(generator)

    yield

    使用了yield的函数被称为生成器。生成器是一个可以返回迭代器的函数,这个生成器函数只能用来做迭代操作,所以生成器可以看作是迭代器。生成器中yield的作用是做暂停操作,并返回一个yield值(这个值用来存储当前的运行信息),在下一次next()时从返回的yield处继续运行。yield支持显示的暂停和恢复,也支持隐式的迭代协议,yield可用在多线程中。

    示例:创建一个斐波那契数列生成器

    import sys def fibonacci(): def fi(n,x,y): for _ in range(n): yield x x,y = y,x+y while True: try: x = int(input('请输入第一个数:')) y = int(input('请输入第二个数:')) n = int(input('请输入生成的数列总数:')) f = fi(n, x, y) while True: try: print(next(f), end=' ') except StopIteration: sys.exit() except ValueError: print('只能输入数字,请重新输入:') fibonacci()

    推导式变生成器

    将外层的括号写成小括号即为一个生成器

    list_generator = (x for x in range(1,21) if x%2==0) for _ in range(5): # 使用循环迭代五次 print(next(list_generator),end=' ')

     

    Processed: 0.048, SQL: 8