@spiritnotes 2016-07-23T15:18:16.000000Z 字数 976 阅读 1494

Python语言实践分析 -- 迭代器协议

Python DOING

迭代器

所谓的迭代器就是实现了迭代器协议的对象，支持循环遍历的操作

for
while
支持迭代的函数，如list、sum....

迭代器协议

容器含有 __iter__() 方法返回对应的迭代器对象
针对迭代器调用其 __next__() 方法返回下一元素， next(iter_)
如果元素已经遍历完全，则产生 StopIteration 异常

>>> i = iter([1, 2, 3, 4])
>>> i
<listiterator object at 0x7fd9be17ffd0>
>>> next(i)
1
>>> next(i)
2
>>> next(i)
3
>>> next(i)
4
>>> next(i)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

而for循环可以对应为如下代码：

for i in l:
    print i
#
iter_ = iter(l)
while True:
    try:
        tmp = iter_.next() # Python3中为iter_.__next\__()
        print tmp
    except StopIteration:
        break

定义迭代器

当自定义迭代器需要定义如下方法：

__iter__
next(Python2)/__next__(ptyhon3)

iter返回self

可以将对象容器定义和iterator定义合并在一起，这样每次迭代器都是返回同样对象。因此所有的迭代器均是对应的同一个迭代器。

iter返回生成器

可以将iter方法定义为yield语句产生的生成器，这样也就不需要再定义next方法了。并且可以支持多次迭代。

iter返回同类新对象

判断对象是否可迭代多次

如果遇到从API获取的迭代器，而使用者可以需要多其多次迭代，则可以通过其迭代是否唯一进行判断。

assert iter(iterator) is not iter(iterator)