Python迭代器
一、迭代器对象(Iterator)
-
迭代器对象
-
指含有_iter_()方法和_next_()方法的方法的类所生成的对象,例如下面所示的对象
-
前面一个方法需要返回对象本身
-
后面一个方法返回下一个数据,如果不存在下一个数据那么需要抛出一个StopIteration异常
class MyIterator(Object): def __init__(self): self.couter = 0 def __iter__(self): #返回自身 return self def __next__(self): self.couter+=1 if(self.couter>=3): #不存在下一个数据则抛出对应异常 raise StopIteration() return self.couter
-
对此迭代器对象,可以通过调用next()方法获取下一个数据,而不是调用__next__方法。
tester = MyIterator() print(next(tester)) #等价于tester.__next__() print(next(tester)) print(next(tester)) #抛出异常
-
for in的执行机理:先调用in对象后面的对象的__iter()__方法返回一个迭代器对象,再不断调用这个返回后对象的_next_()方法,直到该方法抛出StopIteration()异常,最后循环终止
for i in tester: #首先返回自身即tester,再不断调用tester的__next__()方法获取i,直到捕获到异常终止 print(i)
-
二、可迭代对象
-
可迭代对象
-
类中有__iter__方法且返回另外的迭代器对象,例如下面的MyIterable类就是一个可迭代类
class MyIterable(Object): def __init__(self): pass def __iter__(self): return MyIterator() #返回迭代器对象 #other fields TODO here
-
其余的使用和上面的迭代器对象是一致的。例如如果for in循环后面的对象是可迭代对象那么还是调用_iter_()方法返回一个迭代器对象,对这个迭代器对象调用next方法
-