python的迭代器

新的IDE安装完成，是不是迫不及待想要见识见识它的优秀？下面使用pycharm完成我们的第一个类的实现，定制自己的迭代器。

迭代器的特性

迭代器有着跟序列对象一样的行为（当与for一起使用时），但又不同于序列对象，有下面这些特性：

1. 有一个next方法的对象，可以通过next方法获得下一个值，而不能通过索引来获取
2. 当使用next迭代到最后一个元素时将引发StopIteration异常，如果配合for来使用，for将会自动调用next方法和合理的处理StopInteration异常
3. 一次只能拿到一个值，也只有这个值处于内存中
   文件对象就是一个迭代器对象，它的next是readline方法，即一次返回一行数据

定义自己的迭代器

我们在学习编程的时候，相信大家都尝试过用各种语言来实现阶乘，尤其是在讲递归或者循环，又或者算法基础时。下面我们就将阶乘定义成迭代器的形式，我们需要传入一个上限作为参数，迭代器的实现如下：

__author__ = 'anys'
"""
create an iter
an iter must has __iter__ and next method,
raise StopIteration when reach at last item of iter
"""


class Factorial():
    def __init__(self, max_limit):
        self.max = max_limit
        self.value = 1
        self.step = 0

    def __iter__(self):
        return self

    def next(self):
        self.step += 1
        if self.step > self.max:
            raise StopIteration
        self.value *= self.step
        return self.value

根据迭代器的特性，我们定义了next方法，并且在适当的时候（step > max）引发StopIteration异常，实现了__iter__方法仅仅是返回自己。

使用我们的迭代器

使用迭代器就像使用序列对象（列表、元组、字符串）一样简单，

1. 获得迭代器的一个实例
2. 使用该实例的next方法获得一个值，或者使用

for item in iter:
        print item

的方式来获得一个值。下面看一看我们定义的计算阶乘的迭代器，测试代码如下：

if __name__ == '__main__':
    fac = Factorial(20)
    print fac
    print fac.next()
    print fac.next()
    for item in fac:
        print item

测试结果见下图：

如果您有更好的想法，希望多多交流！邮箱：cug_heshun@sina.com