第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']