Python __iter__()迭代器

__iter__():迭代器,生成迭代对象时调用,返回值必须是对象自己,然后for可以循环调用next方法

next():每一次for循环都调用该方法(必须存在)


class A(object):
    def __init__(self,num):
        self.num = num
        self.start_num = -1
    
    def __iter__(self):
        '''
        @summary: 迭代器,生成迭代对象时调用,返回值必须是对象自己,然后for可以循环调用next方法
        '''
        print "__iter__"
        return self
    
    def next(self):
        '''
        @summary: 每一次for循环都调用该方法(必须存在)
        '''
        self.start_num += 1
        if self.start_num >= self.num:
            raise StopIteration()
        return self.start_num
    
if __name__ == "__main__":
    for i in A(10):print i,


输出:
__iter__
0 1 2 3 4 5 6 7 8 9

 

你可能感兴趣的:(python语法总结)