python迭代器资料整理

735 查看

什么是迭代器

相关概念定义

迭代器(Iterator):

满足迭代协议的对象就是迭代器
iterator就是实现了Iteration Protocol的对象,这类对象都支持循环遍历的操作(for/while/支持迭代的函数list() sum()…)。

迭代协议(Iteration Protocol):

内建函数iter()接收一个可迭代对象,并返回一个可迭代对象.
每次将这个可迭代对象传递给next()函数,都会返回它所包含的下一个元素,当迭代完最后一个元素时,就会触发StopIteration异常。

满足以上要求的对象,就是迭代器。

迭代隐含的操作

在每次的迭代语句中,python都会按照迭代协议去对迭代器进行迭代。其实,在实际执行中,python会进行一些其他的操作:

  1. 将需要迭代的对象作为参数传递给iter函数
  2. iter返回一个迭代器对象
  3. 每次循环则将返回的迭代器对象传递给next函数
  4. 循环至最后一个元素,触发StopIteration

for语句为例
当我们在Python中执行循环语句for i in foo的时候,其背后的操作是:

  1. foo = iter(foo)
  2. next(foo)

next(foo)在python3中执行的是:foo.__next__(),在python2中则是:foo.next()

迭代器的实现

迭代器是用class来实现的。其中必需实现的有两个方法:__iter__next(python2)/__next__(python3)。其中,__iter__必需返回一个迭代器对象,next则负责迭代逻辑并在迭代完毕时触发异常。

如下:


迭代器的特殊情况

迭代器的__iter__返回self

迭代器的__iter__方法需要返回的是一个具有next方法的可迭代对象。如果当__iter__返回的是self的话,就会产生其他意想不到的效果。

执行结果: