新的IDE安装完成,是不是迫不及待想要见识见识它的优秀?下面使用pycharm完成我们的第一个类的实现,定制自己的迭代器。
迭代器的特性
迭代器有着跟序列对象一样的行为(当与for一起使用时),但又不同于序列对象,有下面这些特性:
- 有一个next方法的对象,可以通过next方法获得下一个值,而不能通过索引来获取
- 当使用next迭代到最后一个元素时将引发StopIteration异常,如果配合for来使用,for将会自动调用next方法和合理的处理StopInteration异常
- 一次只能拿到一个值,也只有这个值处于内存中
文件对象就是一个迭代器对象,它的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__方法仅仅是返回自己。
使用我们的迭代器
使用迭代器就像使用序列对象(列表、元组、字符串)一样简单,
- 获得迭代器的一个实例
- 使用该实例的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