标签：
对于一个列表，a = [1, 2, 3, 4]，我们最常见的遍历方式就是：
a = [1, 2, 3, 4] for item in a: print item
这里我们研究一种新的方式，就是迭代器。
在C++的STL中大量使用了迭代器，迭代器的作用当然就是遍历容器中的元素，而且他的好处就在于分离了容器的实现和遍历操作，不管我们使用什么类型的容器，使用迭代器的操作几乎是如出一辙。
看下面的代码：
>>> a = [2, 3, 4] >>> it = iter(a) >>> print it.next() 2 >>> print it.next() 3 >>> print it.next() 4 >>> print it.next() Traceback (most recent call last): File "<stdin>", line 1, in <module> StopIteration >>>
在上面的代码中，iter函数创建了一个可以迭代的对象，然后每次调用next方法，都能从其中取出元素。当没有元素可以迭代时，便抛出一个异常StopIteration。
所以我们上面的for循环可以这样改写：
a = [1, 2, 3, 4] it = iter(a) item = None while True: try: item = it.next() except StopIteration: break print item #do_something
如何创建迭代器
现在我们想对我们自定义的class进行迭代操作，应该怎么办？
这里的关键是实现__iter__和next两个函数。
#!/usr/bin/env python #coding: utf-8 class IterList: def __init__(self, elem): self.iter = iter(elem) def __iter__(self): return self def next(self): return self.iter.next() if __name__ == ‘__main__‘: a = [1, 2, 3, 4] test = IterList(a) for item in test: print item
这里我们仅仅是对class内部持有的元素做了一个包装，我们的__iter__返回的是自身，next则是调用的存储的iter的next方法。
现在我们提供一个稍微复杂的版本，这个版本可以允许向next函数传递参数，指定取出几个值。
#!/usr/bin/env python #coding: utf-8 class IterList: def __init__(self, elem): self.iter = iter(elem) def __iter__(self): return self def next(self, howmany=1): result = [] for i in range(howmany): try: result.append(self.iter.next()) except StopIteration: raise return result if __name__ == ‘__main__‘: s = range(20) test = IterList(s) print test.next() print test.next() print test.next(3)
标签：
原文地址：http://www.cnblogs.com/inevermore/p/4216848.html
