[Python知识库]python 迭代器、生成器（通俗易懂！）

引言

前几天面试，被问到python 迭代器，并用它输出一个斐波那契数列。当场懵逼。。。于是乎，想彻底搞明白python的迭代器，找了很多文章，都没能让我“茅塞顿开、恍然大悟”，知道看来B站的一个视频讲解，才终于懂了！视频

正文

可迭代对象、迭代器、生成器的关系：

准确的说，它们仨是包含关系：

按照 迭代器–>可迭代对象–>生成器 的顺序讲解（方便理解！）

迭代器（Iterator）

包含有2个关键函数：__ iter__()和__next__()，其实起作用的只有__next__()。每次调用一次__next__()，迭代器对象会遍历“下一个元素”，并返回。

可迭代对象（Iterable）

我给它取了个外号——产生迭代器对象的工厂。
包含1个关键函数：__ iter__()，没错，上面的迭代器也有，因此迭代器也是一种可迭代对象。
__ iter__()作用：用来返回一个迭代器。

下面根据代码，比较两者。

class MyList(object):            # 可迭代对象类
    def __iter__(self):
        return MyListIterator()  # 返回该可迭代对象的迭代器类的实例
 
 
class MyListIterator(object):    # 迭代器类
 
    def __init__(self):
        self.count= 0            # 当前迭代值，初始为0
 
    def __iter__(self):
        return self              # 返回该对象的迭代器类的实例；因为自己就是迭代器，所以返回self
 
    def __next__(self):          # 迭代器类必须实现的方法
        while self.count < 3:
            self.count += 1
            return self.count-1  # 返回当前迭代值
        raise StopIteration      # 超出边界，抛出异常
 

• MyList：可迭代对象类【即：一个产生迭代器对象的工厂】，返回了一个迭代器——MyListIterator()。
• MyListIterator：迭代器类，可以用__next__()去完成遍历。同时，（敲黑板啦！！！），它也是一个可迭代对象类，返回了一个迭代器——self【他自己】。

小结

迭代器对象可以理解为：一个具备“遍历”功能的工厂！

生成器

class MyGenerator():
	yield 1
	yield 2
	
obj = MyGenerator()
for item in obj:
	print(item)	

疑惑：为什么他没有__next__()，它反而是迭代器类呢？
解答：obj = MyGenerator() 是创建生成器对象（因为有yield，obj内部是根据生成器类generator类创建的对象），生成器类的内部也声明了__iter__()和__next__()函数。可以通过dir(obj)再命令行查看obj对象包含的函数。

for循环

有必要提一下for循环的原理。分为两步。

1. 首先调用__iter__()函数，即发挥了 可迭代对象 的【工厂】功能，产生一个迭代器对象。仅仅开始时执行一次。

2. 然后，下面的遍历工作，交给生成的 迭代器，即每一轮执行一次__next__()函数。

总结

这篇文章是针对迭代器的概念，做一个简明扼要的介绍，让大家有个直观的了解。可以观看B站的视频进行学习。至于详细用法，就不讲了。

回到引言

完成我的最初的使命：使用迭代器，生成斐波那契数列。【代码出处，侵删】

class Fibonacci(object):
    """斐波那契数列得迭代器"""
    def __init__(self,nums):
        self.nums = nums   # 传入参数，生成斐波那契数列的个数
        self.a = 0   
        self.b = 1
        self.i =0    # 用于记忆生成的个数
    def __iter__(self):
        return self
 
    def __next__(self):
           
       ret = self.a   # 记忆第一个数
 
       if self.i < self.nums:
            self.a, self.b = self.b,self.a +self.b
            self.i += 1
            return ret
       else:
           raise StopIteration   # 停止迭代
 
nums = int(input("请输入需要生成Fibonacci数列项的个数："))
fobo = Fibonacci(nums)
 
for num in fobo:
    print(num)