迭代器
可迭代对象
list,str,tuple,etc.—> for …in…—>遍历
迭代器协议:对象必须提供一个next方法,执行该方法要么返回迭代中 得下一项,要么引起StopIteration,以终止迭代(只往下走,不后退)
所以,实现迭代器协议的对象就是可迭代对象。
如何实现?
- 通过在对象内部定义一个
__iter__ 方法
li=[1,2,3]
sr='abcd'
print(li.__iter__()) #---->iter(li)
li_iter=li.__iter__()
print(li_iter.__next__())#--->next(li_iter)
print(li_iter.__next__())
print(next(li_iter))
1
2
3
Traceback (most recent call last):
File “D:/pycharm/PyCharm 2018.1/untitled/start study/2019 7/teach_class/day0730.py”, line 8, in
print(next(li_iter))
StopIteration
for i in 100:
print(i)
TypeError: ‘int’ object is not iterable
from collections.abc import Iterable
print(isinstance([],Iterable))
print(isinstance(str(),Iterable))
判断对象是否是可迭代的。
自定义一个类,它可以容纳数据,
from collections.abc import Iterable
class MyClass:
def __init__(self):
self.names=[]
def add(self,name):
self.names.append(name)
def __iter__(self):
return None
my_class=MyClass()
my_class.add('tom')
my_class.add('Jack')
my_class.add('Lucy')
print(my_class.names)
print(isinstance(my_class,Iterable))
[‘tom’, ‘Jack’, ‘Lucy’]
True
在迭代一个可迭代对象的时候,实际上就是获取该对象的一个迭代器,然后通过迭代器一次获取对象的每一个数据。
for…in… 循环的本质就是通过Iter()函数获取可迭代对象的Iterable的迭代器,然后对获取到的迭代器不断调用next()方法来获取下一个值并将其赋值,当遇到StopIteration的异常后,退出。
应用场景
迭代器的核心就是通过next()函数调用返回下一个数据值,如果每次返回的数据值不是在一个已有的数据集合中读取的,而是通过程序按照一定规律计算生成,那就意味着可以不用依赖一个已有的数据集合,namely,无需将所有迭代对象数据一次性缓存下来供后续使用。这样,可以节省大量的存储(内存)空间。
demo:
斐波那契数列
for…in…的方式来遍历斐波那契数列中的前n个数。
通过迭代来实现,每次迭代都可以通过数学计算生成下一个数
class FIBiterato(object):
def __init__(self,n):
self.n=n #记录生成的斐波那契数列
self.current_index=0 #记录当前记录的索引
self.num1=0
self.num2=1 #定义两个初始值
def __next__(self): #调用next函数来获取下一个数
if self.current_index<self.n:
num=self.num1
self.num1,self.num2=self.num2,self.num1+self.num2
self.current_index+=1
return num
else:
raise StopIteration
def __iter__(self):
return self
fib=FIBiterato(10)
for num in fib:
print(num,end=' ')
n
def __iter__(self):
return self
fib=FIBiterato(10)
for num in fib:
print(num,end=' ')