python的迭代器

来源:互联网 发布:学生赚软件下载 编辑:程序博客网 时间:2024/06/05 05:10

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

迭代器的特性

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

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

定义自己的迭代器

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

__author__ = 'anys'"""create an iteran 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

0 0
原创粉丝点击