第13章 面向对象编程(13)

13.13.3 迭代器(RandSeq和AnyIter)

1.RandSeq

>>> from random import choice >>> class RandSeq(object): ... def __init__(self, seq): ... self.data = seq ... def __iter__(self): ... return self ... def next(self): ... return choice(self.data) ... >>>
__init__()方法执行前述的赋值操作。__iter__()仅返回self，这就是如何将一个对象声明为迭代器的方式，最后，调用next()来得到迭代器中连续的值。这个迭代器唯一的亮点是它没有终点。

2. AnyIter

对一个对象调用iter()就可以得到它的迭代器。

>>> class AnyIter(object): ... def __init__(self, data, safe = False): ... self.iter = iter(data) ... self.safe = safe ... def __iter__(self): ... return self ... def next(self, howmany = 1): ... retval = [] ... for eachitem in range(howmany): ... try: ... retval.append(self.iter.next()) ... except StopIteration: ... if self.safe: ... break ... else: ... raise ... return retval ... >>> >>> for i in AnyIter('abc'): ... print i ... ['a'] ['b'] ['c'] >>> for i in AnyIter('abc', True): ... print i>>> iter(AnyIter('ad')).next(5) Traceback (most recent call last): File "<interactive input>", line 1, in <module> File "<interactive input>", line 11, in next StopIteration >>> iter(AnyIter('adc')).next(2) ['a', 'd']