首页 > 代码库 > python高级编程-Part1 生成器和迭代器
python高级编程-Part1 生成器和迭代器
迭代器和生成器是python学者们经常谈到的话题,我也不能免俗,因为实在值得总结一下。
迭代器
迭代器是对可迭代对象进行操作,通过next方法一次吐出一个元素的工具。我们用到的for..in..内部使用的就是迭代器功能。
如果要自定义一个迭代器类的话,需要满足下面的条件:
需要在类中定义__iter__方法返回self自身,表示这是一个迭代器;
需要定义next方法来返回迭代的值,其中应该包含StopIteration异常的判断
下面试着写一个自定义迭代器类的例子(模仿自Python高级编程一书):
class CustomIter(object): def __init__(self, step): self.step = step def __iter__(self): return self def next(self): if self.step == 0: return StopIteration self.step -=1 return self.step i = CustomIter(2) #此处如果换为for i in CustomIter(2),则不需要后续的next手工调用, #也不会出现StopIteration的异常,因为for循环内部实现了next调用 print i.next() print i.next() print i.next() 运行结果如下: 1 0 <type ‘exceptions.StopIteration‘>
迭代器实际是一种底层的特性和概念,但是为生成器提供了一种支持。
生成器
有记忆性的函数暂停和恢复执行的表现,使用yield替代return语句,表示每次返回一个值并暂停执行。当再次被外部next调用时,利用函数上下文自动从暂停的位置继续运行,直到再次遇到yield语句。
让我们从代码中理解上面没太听明白的话:
#!/usr/bin/env python #encoding:utf-8 def testGenerator(): for x in ("first", "second", "third"): print "this is " + x yield x #每次返回x之后,会停止 print "do other things" b = testGenerator() print b.next() print u"\n------再次执行next():------\n" #再次运行next print b.next() 运行结果如下: this is first first ------再次执行next():------ do other things this is second second
生成器的其他形式:
yield表达式,即在函数体中存在var = (yield),表示可以得到客户端调用send(生成器对象的方法)发送进来的数据
后面再说吧,头疼
本文出自 “无名” 博客,请务必保留此出处http://xdzw608.blog.51cto.com/4812210/1601317
python高级编程-Part1 生成器和迭代器
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。