现在的位置: 首页 > 综合 > 正文

开发自定义python 迭代器

2013年09月01日 ⁄ 综合 ⁄ 共 319字 ⁄ 字号 评论关闭

     

class test(object):
    def __init__(self,data_iter,stop):
        self.data_iter = data_iter
        self.stop = stop
        self.start = 0

    def __iter__(self):
        return self

    def next(self):
        self.start += 1
        if self.start >= self.stop:
            raise StopIteration
        return self.data_iter.next() + str(self.start)

if __name__ == "__main__":
    f = open('part-00000','r')
    t = test(f,12)
    for i in t:
      print i

抱歉!评论已关闭.