python-迭代器与生成器_头歌迭代器和生成器-CSDN博客

本文链接：https://blog.csdn.net/JBY2020/article/details/108955842

迭代器

迭代器

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

迭代器的创建与使用

创建迭代器：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=' ')