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输出一个杨辉三角的例子
Jun 13 Python
谈谈Python进行验证码识别的一些想法
Jan 25 Python
Python中装饰器高级用法详解
Dec 25 Python
python中实现将多个print输出合成一个数组
Apr 19 Python
Python中使用Counter进行字典创建以及key数量统计的方法
Jul 06 Python
基于python历史天气采集的分析
Feb 14 Python
Python环境Pillow( PIL )图像处理工具使用解析
Sep 12 Python
python将dict中的unicode打印成中文实例
May 11 Python
Python可以实现栈的结构吗
May 27 Python
python 监控logcat关键字功能
Sep 04 Python
详解Python流程控制语句
Oct 28 Python
python中sqllite插入numpy数组到数据库的实现方法
Jun 21 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
用ODBC的分页显示
2006/10/09 PHP
php mysql数据库操作分页类
2008/06/04 PHP
javascript处理table表格的代码
2010/12/06 Javascript
基于JQuery的日期联动实现代码
2011/02/24 Javascript
JavaScript中的原型和继承详解(图文)
2014/07/18 Javascript
jquery 获取 outerHtml 包含当前节点本身的代码
2014/10/30 Javascript
什么是 AngularJS?AngularJS简介
2014/12/06 Javascript
jQuery应用之jQuery链用法实例
2015/01/19 Javascript
JS DOMReady事件的六种实现方法总结
2016/11/23 Javascript
微信小程序 input输入框控件详解及实例(多种示例)
2016/12/14 Javascript
微信小程序tabbar不显示解决办法
2017/06/08 Javascript
写给vue新手们的vue渲染页面教程
2017/09/01 Javascript
vue中配置mint-ui报css错误问题的解决方法
2017/10/11 Javascript
使用vue打包时vendor文件过大或者是app.js文件很大的问题
2018/06/29 Javascript
详解vue-element Tree树形控件填坑路
2019/03/26 Javascript
[57:29]Alliance vs KG 2019国际邀请赛小组赛 BO2 第二场 8.16
2019/08/17 DOTA
[02:13] 完美世界DOTA2联赛PWL DAY5集锦
2020/11/03 DOTA
python 根据正则表达式提取指定的内容实例详解
2016/12/04 Python
Django 实现购物车功能的示例代码
2018/10/08 Python
Python实现字典排序、按照list中字典的某个key排序的方法示例
2018/12/18 Python
python读取txt文件中特定位置字符的方法
2018/12/24 Python
Python 仅获取响应头, 不获取实体的实例
2019/08/21 Python
python对象转字典的两种实现方式示例
2019/11/07 Python
python中栈的原理及实现方法示例
2019/11/27 Python
python如何基于redis实现ip代理池
2020/01/17 Python
python获取栅格点和面值的实现
2020/03/10 Python
DataFrame.groupby()所见的各种用法详解
2020/06/14 Python
CSS3中Transform动画属性用法详解
2016/07/04 HTML / CSS
自荐书范文
2013/12/08 职场文书
应用艺术专业个人的自我评价
2014/01/03 职场文书
小学后勤管理制度
2014/01/14 职场文书
酒店个人求职信范文
2014/01/25 职场文书
开展党的群众路线教育实践活动方案
2014/02/05 职场文书
2014幼儿教师个人工作总结
2014/12/03 职场文书
死亡诗社观后感
2015/06/05 职场文书
PyTorch梯度裁剪避免训练loss nan的操作
2021/05/24 Python