引言
前几天面试,被问到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循环的原理。分为两步。
-
首先调用__iter__()函数,即发挥了 可迭代对象 的【工厂】功能,产生一个迭代器。仅仅在最开始时执行一次。
-
然后,下面的遍历工作,交给生成的 迭代器,即每一轮执行一次__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)