3. Python3 中的迭代器

  • 迭代器的定义:具有__next__(或者next,python2)方法的对象。
  • 自动可迭代的迭代器:实现__iter__方法的迭代器。通常迭代器是指自动可迭代的迭代器
  • 迭代器的优势:相比使用List迭代,如果迭代数据量很大,List方法将消耗大量内存,而迭代器可以节约不少内存。获取速度和内存的提升。
  • 其他优势:简单、通用、优雅
  • 迭代器的终止:采用异常机制,在__next__中无法提供下一个值的时候,抛出 raise StopIteration 即可。
  • 迭代器的使用技巧:a) 直接到元组,链表;b)生成一次,只使用一次。当然重新初始化也是可行的,不建议用。

失败的版本

以 Fibs 数列为例子。

>>> class Fibs():
...     def __init__(self):
...         self.a = 0
...         self.b = 1
...     def __next__(self):
...         self.a, self.b = self.b, self.a+self.b
...         return self.a
... 
>>> fibs = Fibs()
>>> print(fibs.__next__(),fibs.__next__(),fibs.__next__(),fibs.__next__(),fibs.__next__(),fibs.__next__())
1 1 2 3 5 8
>>> for f in fibs:
...     print(f)
...     if (f>100):
...         break
... 
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: 'Fibs' object is not iterable

错误提示为:Fibs的对象确实是迭代器(能够手动通过__next__给出迭代结果),但是不可自动迭代。尽管我们已经有了__next__的方法,却没有实现__iter__方法。

成功的版本

>>> class Fibs():
...     def __init__(self):
...         self.a = 0
...         self.b = 1
...     def __next__(self):
...         self.a, self.b = self.b, self.a+self.b
...         return self.a
...     def __iter__(self):
...         return self
... 
>>> fibs = Fibs()
>>> print(fibs.__next__(),fibs.__next__(),fibs.__next__(),fibs.__next__(),fibs.__next__(),fibs.__next__())
1 1 2 3 5 8
>>> for f in fibs:
...     print(f)
...     if (f>100):
...         break
... 
13 21 34 55 89 144

添加了__iter__函数,该迭代器可以自动迭代了。注意,自动迭代是在手动迭代的基础上继续进行的。

  • 迭代器的终止
    上述例子是通过 if语句主动终止的,如果别人使用你的迭代器,不主动终止,那么肯定就爆了(我已经重新启动过电脑一次了)。主动的终止方式是采用异常 raise StopIteration 即可
>>> class Fibs():
...     def __init__(self):
...         self.a = 0
...         self.b = 1
...     def __next__(self):
...         self.a, self.b = self.b, self.a+self.b
...         if self.a > 100: raise StopIteration
...         return self.a
...     def __iter__(self):
...         return self
... 
>>> fibs = Fibs()
>>> for f in fibs:
...     print(f)
... 
1 1 2 3 5 8 13 21 34 55 89

这样便不需要额外的stop停止语句。

从迭代器到序列

直接使用truple、list函数转换

>>> class Fibs():
...     def __init__(self):
...         self.a = 0
...         self.b = 1
...     def __next__(self):
...         self.a, self.b = self.b, self.a+self.b
...         if self.a > 100: raise StopIteration
...         return self.a
...     def __iter__(self):
...         return self
... 
>>> fibs = Fibs()
>>> print(tuple(fibs))
(1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89)
>>> fibs = Fibs()
>>> print(list(fibs))
[1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89]

注意: 迭代器用完了之后,需要重置,不然得不到你要的结果。最佳的使用方式是用一次,建立一次。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,607评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,047评论 2 379
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,496评论 0 335
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,405评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,400评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,479评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,883评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,535评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,743评论 1 295
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,544评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,612评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,309评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,881评论 3 306
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,891评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,136评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,783评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,316评论 2 342

推荐阅读更多精彩内容

  • 迭代器(Iterable) 简单来说,迭代器对象(my_list)可以让以下代码正常工作: 如果对象实现了__it...
    SOLAREST阅读 538评论 0 1
  • PYTHON-进阶-ITERTOOLS模块小结转自wklken:http://wklken.me/posts/20...
    C_Y_阅读 951评论 0 2
  • 对于list、string、tuple、dict等这些容器对象,使用for循环遍历是很方便的。在后台for语句对容...
    菜鸟辣妈阅读 2,396评论 0 1
  • “老板,还有蛋糕吗?” “没了。” “哦,那算了。” “你什么时候要?” “现在。” “啊?” “现在!” “好,...
    我是大脸喵呀阅读 1,619评论 2 1
  • 矗立在戈壁 默默无语也风流 一段奇景千古苍凉 铸就一生绝唱 留下那蓬勃的沧桑 是你 装扮了这片亘古寂灭的土地 召唤...
    瀚正阅读 410评论 0 4