Python的迭代器和生成器


Posted in Python onJuly 29, 2015

先说迭代器,对于string、list、dict、tuple等这类容器对象,使用for循环遍历是很方便的。在后台for语句对容器对象调用iter()函数,iter()是python的内置函数。iter()会返回一个定义了next()方法的迭代器对象,它在容器中逐个访问容器内元素,next()也是python的内置函数。在没有后续元素时,next()会抛出一个StopIteration异常,通知for语句循环结束。比如:

>>> s = 'abc'
>>> it = iter(s)
>>> it
<str_iterator object at 0x7f71fefe9d68>
>>> next(it)
'a'
>>> next(it)
'b'
>>> next(it)
'c'
>>> next(it)
Traceback (most recent call last):
 File "<stdin>", line 1, in <module>
StopIteration

上面说的都是python自带的容器对象,它们都实现了相应的迭代器方法,那如果是自定义类需要遍历怎么办?方法很简单,对这个类AClass,实现一个__iter__(self)方法,使其返回一个带有__next__(self)方法的对象就可以了。如果你在AClass刚好也定义了__next__(self)方法(一般使用迭代器都会定义),那在__iter__里只要返回self就可以。废话少说,先上代码:

class Fib(object):
  def __init__(self, max):
    super(Fib, self).__init__()
    self.max = max

  def __iter__(self):
    self.a = 0
    self.b = 1
    return self

  def __next__(self):
    fib = self.a
    if fib > self.max:
      raise StopIteration
    self.a, self.b = self.b, self.a + self.b
    return fib

def main():
  fib = Fib(100)
  for i in fib:
    print(i)

if __name__ == '__main__':
  main()

简单讲下代码会干什么,定义了一个Fib类,用于生成fibonacci序列。用for遍历时会逐个打印生成的fibonacci数,max是生成的fibonacci序列中数字大小的上限。

在类的实现中,定义了一个__iter__(self)方法,这个方法是在遍历时被iter()调用,返回一个迭代器。因为在遍历的时候,是直接调用的python内置函数iter(),由iter()通过调用__iter__(self)获得对象的迭代器。有了迭代器,就可以逐个遍历元素了。而逐个遍历的时候,也是使用内置的next()函数通过调用对象的__next__(self)方法对迭代器对象进行遍历。所以要实现__iter__(self)和__next__(self)。而且因为实现了__next__(self),所以在实现__iter__(self)的时候,直接返回self就可以。

为了更好理解,我再简单重复下上面说的那一段:在循环遍历自定义容器对象时,会使用python内置函数iter()调用遍历对象的__iter__(self)获得一个迭代器,之后再循环对这个迭代器使用next()调用迭代器对象的__next__(self)。__iter__只会被调用一次,而__next__会被调用 n 次。

下面说生成器。

生成器(Generator)是创建迭代器的简单而强大的工具。它们写起来就像是正规的函数,只是在需要返回数据的时候使用yield语句。每次next()被调用时,生成器会返回它脱离的位置(它记忆语句最后一次执行的位置和所有的数据值)。以下示例演示了生成器可以很简单的创建出来:

>>> def reverse(data):
...   for index in range(len(data)-1, -1, -1):
...     yield data[index]
... 
>>> for char in reverse('hello'):
...   print(char)
... 
o
l
l
e
h

关于迭代器和生成器的区别,生成器能做到迭代器能做的所有事,而且因为自动创建了__iter__()和 next()方法,生成器显得特别简洁,而且生成器也是高效的。除了创建和保存程序状态的自动方法,当发生器终结时,还会自动抛出StopIteration异常。一个带有yield的函数就是一个 生成器,它和普通函数不同,生成一个 generator 看起来像函数调用,但不会执行任何函数代码,直到对其调用next()(在 for 循环中会自动调用next())才开始执行。虽然执行流程仍按函数的流程执行,但每执行到一个yield语句就会中断,并返回一个迭代值,下次执行时从yield的下一个语句继续执行。看起来就好像一个函数在正常执行的过程中被yield中断了数次,每次中断都会通过yield返回当前的迭代值(yield暂停一个函数,next()从其暂停处恢复其运行)。

另外对于生成器,python还提供了一个生成器表达式:类似与一个yield值的匿名函数。表达式本身看起来像列表推到, 但不是用方括号而是用圆括号包围起来:

>>> unique_characters = {'E', 'D', 'M', 'O', 'N', 'S', 'R', 'Y'}
>>> gen = (ord(c) for c in unique_characters)
>>> gen
<generator object <genexpr> at 0x7f2be4668678>
>>> for i in gen:
...   print(i)
... 
69
79
83
77
82
78
89
68
>>>

如果需要,可以将生成器表达式传给tuple、list或是set来迭代所有的值并且返回元组、列表或是集合。在这种情况下,不需要一对额外的括号 ———— 直接将生成器表达式 ord(c) for c in unique_characters传给tuple()等函数就可以了, Python 会推断出它是一个生成器表达式。

最后,为什么要使用生成器?因为效率。使用生成器表达式取代列表解析可以同时节省 cpu 和 内存(ram)。如果你构造一个列表的目的仅仅是传递给别的函数,(比如 传递给tuple()或者set()), 那就用生成器表达式替代吧!

以上所述就是本文的全部内容了,希望大家能够喜欢。

Python 相关文章推荐
python使用cPickle模块序列化实例
Sep 25 Python
使用Pyrex来扩展和加速Python程序的教程
Apr 13 Python
深入学习Python中的装饰器使用
Jun 20 Python
Python实现的中国剩余定理算法示例
Aug 05 Python
python表格存取的方法
Mar 07 Python
Window10+Python3.5安装opencv的教程推荐
Apr 02 Python
python pandas 对series和dataframe的重置索引reindex方法
Jun 07 Python
Python中flatten( )函数及函数用法详解
Nov 02 Python
Python中作用域的深入讲解
Dec 10 Python
python 将对象设置为可迭代的两种实现方法
Jan 21 Python
详解python持久化文件读写
Apr 06 Python
python实现遍历文件夹图片并重命名
Mar 23 Python
在Python程序中操作MySQL的基本方法
Jul 29 #Python
Python操作Word批量生成文章的方法
Jul 28 #Python
Python实现批量转换文件编码的方法
Jul 28 #Python
Python中subprocess的简单使用示例
Jul 28 #Python
Python中文竖排显示的方法
Jul 28 #Python
Python中的getopt函数使用详解
Jul 28 #Python
Python3访问并下载网页内容的方法
Jul 28 #Python
You might like
Codeigniter整合Tank Auth权限类库详解
2014/06/12 PHP
Thinkphp将二维数组变为标签适用的一维数组方法总结
2014/10/30 PHP
[原创]php简单隔行变色功能实现代码
2016/07/09 PHP
php成功操作redis cluster集群的实例教程
2019/01/13 PHP
js控制不同的时间段显示不同的css样式的实例代码
2013/11/04 Javascript
动态创建script在IE中缓存js文件时导致编码的解决方法
2014/05/04 Javascript
javascript函数式编程实例分析
2015/04/25 Javascript
AngularJS入门教程之控制器详解
2016/07/27 Javascript
javascript验证内容为数字以及长度为10的简单实例
2016/08/20 Javascript
js中setTimeout的妙用--防止循环超时
2017/03/06 Javascript
Vue.js实现一个漂亮、灵活、可复用的提示组件示例
2017/03/17 Javascript
9种改善AngularJS性能的方法
2017/11/28 Javascript
react-router v4如何使用history控制路由跳转详解
2018/01/09 Javascript
基于vue-router 多级路由redirect 重定向的问题
2018/09/03 Javascript
小程序实现自定义导航栏适配完美版
2019/04/02 Javascript
vue实现Input输入框模糊查询方法
2021/01/29 Javascript
ES6新增的数组知识实例小结
2020/05/23 Javascript
javascript实现贪吃蛇小游戏
2020/07/28 Javascript
Python函数式编程指南(一):函数式编程概述
2015/06/24 Python
Python单向链表和双向链表原理与用法实例详解
2018/08/31 Python
python读写csv文件并增加行列的实例代码
2019/08/01 Python
浅谈Python中文件夹和python package包的区别
2020/06/01 Python
猎人靴英国官网:Hunter Boots
2017/02/02 全球购物
定制iPhone和Macbook保护壳:Slick Case
2018/11/21 全球购物
Vinatis德国:法国领先的葡萄酒邮购公司
2020/09/07 全球购物
华为c/c++笔试题
2016/01/25 面试题
物业总经理助理岗位职责
2014/06/29 职场文书
环境保护与污染治理求职信
2014/07/16 职场文书
党的群众路线教育实践活动个人对照检查材料(校长)
2014/11/05 职场文书
销售经理岗位职责
2015/01/31 职场文书
个性发展自我评价2015
2015/03/09 职场文书
面试通知短信
2015/04/20 职场文书
少先队中队工作总结
2015/08/14 职场文书
导游词之宁夏贺兰山岩画
2019/11/08 职场文书
SQL注入详解及防范方法
2021/12/06 MySQL
HTML基本元素标签介绍
2022/02/28 HTML / CSS