Python 3.x自定义迭代器对象

24K纯开源

发布于 2018-01-18 10:11:42

6080

发布于 2018-01-18 10:11:42

Python 3.x与Python 2.x之间存在着较多的语法细节差异。今天在看Python核心编程的时候，说到了自定义迭代器对象。于是动手将源码打了一遍，原书代码如下：

class AnyIter(object):
    def __init__(self, data, safe=False):
        """ The initialization of iterators """
        self.safe = safe
        self.iter = iter(data)
    def __iter__(self):
        """ return a iterator """
        return self
    def next(self, count=1):
        """ Return arbitrary numbers of elements """
        retval = []
        for item in range(count):
            try:
                retval.append(self.iter.next())
            except StopIteration:
                if self.safe:
                    break
                else:
                    raise    # reraise the exception again 
        return retval

if __name__ == '__main__':
    a = AnyIter(range(10), True)
    b = iter(a)
    for item in range(1,5):
        print('{}:{}'.format(item, a.next(item)))

我机器上装的是Python 3.3.2，在控制台运行该脚本的时候直接抛出异常TypeError: