生成器和迭代器 Python

生成器 generator

An object created by a generator function.
Each yield temporarily suspends processing, remembering the location execution state (including local variables and pending try-statements). When the generator iterator resumes, it picks up where it left off (in contrast to functions which start fresh on every invocation).

循环的过程中计算。每次循环都会产生返回值(通过yield可以抛出),极大地减少需要的内存。

生成器表达式

生成器表达式是迭代和列表解析的组合。
在生成一个列表[1,2,3,4,5,6,7,8,9,10]的时候:

# 列表推导生成式
liist1 = [i+1 for i in range(10)]

列表推导生成式可以直接生成列表并且储存所有数据 [1,2,3,4,5,6,7,8,9,10]

# 生成器
liist2 = (i+1 for i in range(10))

生成器则是生成了 <generator object <genexpr> at 0x00F02CA0>
逐个输出可以不断通过next(liist2)函数获得生成器的下一个返回值。在没有更多的元素可以生成时,会抛出StopIteration 错误。这种做法并不多见。
generator 也是一个可迭代对象,所以我们可以通过 for loop循环输出。

for i in liist2:
    print(i)

在这里 for loop 实际上就是通过不断调用next()函数实现的。等价于:

while True:
    try:
        x = next(liist2)
    # 遇到StopIteration就退出循环    
    except StopIteration:
        break

生成器函数

生成器函数会在每次执行完毕后利用 yield 自动挂起,并输出一个返回值,之后随时可以重新利用.__next__()继续执行,直至最后没有返回值引起StopIteration异常,结束迭代。

  • 斐波那契数列
    在生成斐波那契数列时,如果想要输出每一个数, 每次计算后print()是一种方式:
#fibonacci数列1,1,2,3,5,8,12,21,34.
def fibo1(n):
    a,b = 0,1
    count = 0
    while count < n:
        a,b =b,a+b
        count += 1
        print(a)
    return 'Done!'

fibo1(10)

也可以调用 yield 完成:

#fibonacci数列1,1,2,3,5,8,12,21,34.
def fibo2(n):
    a,b = 0,1
    count = 0
    while count < n:
        yield b
        a,b =b,a+b
        count += 1
    return 'Done!'

res = fibo2(10)
for i in res:
    print(i)

使用生成器完成的好处是可以减少内存占用,每次运算时用多少占多少。

  • 单线程并发
    yield 也可以单线程实现并发运算效果:
import time
def consumer(name):
    print(f"{name} 准备讲课啦!")
    while True:
       lesson = yield
 
       print("开始[%s]了,[%s]老师来讲课了!" %(lesson,name))
 
 
def producer():
    c = consumer('Tom')
    c2 = consumer('Jerry')
    c.__next__()
    c2.__next__()

    print("同学们开始上课 了!")
    tut = ['math','english','chinese','salksjmdlaksjd']
    for i in tut:
        time.sleep(1)
        print("到了两个同学!")
        c.send(i)
        c2.send(i)

producer()

输出:

Tom 准备讲课啦!
Jerry 准备讲课啦!
同学们开始上课 了!
到了两个同学!
开始[math]了,[Tom]老师来讲课了!
开始[math]了,[Jerry]老师来讲课了!
到了两个同学!
开始[english]了,[Tom]老师来讲课了!
开始[english]了,[Jerry]老师来讲课了!
到了两个同学!
开始[chinese]了,[Tom]老师来讲课了!
开始[chinese]了,[Jerry]老师来讲课了!
到了两个同学!
开始[salksjmdlaksjd]了,[Tom]老师来讲课了!
开始[salksjmdlaksjd]了,[Jerry]老师来讲课了!

可迭代的 Iterable 和 迭代器 Iterator

iterables

An object capable of returning its members one at a time. Examples of iterables include all sequence types (such as list, str, and tuple) and some non-sequence types like dict, file objects, and objects of any classes you define with an __iter__() method or with a __getitem__() method that implements Sequence semantics.
Iterables can be used in a for loop and in many other places where a sequence is needed (zip(), map(), …). When an iterable object is passed as an argument to the built-in function iter(), it returns an iterator for the object. This iterator is good for one pass over the set of values. When using iterables, it is usually not necessary to call iter() or deal with iterator objects yourself. The for statement does that automatically for you, creating a temporary unnamed variable to hold the iterator for the duration of the loop. See also iterator, sequence, and generator.

iterator

An object representing a stream of data. Repeated calls to the iterator’s __next__() method (or passing it to the built-in function next()) return successive items in the stream. When no more data are available a StopIteration exception is raised instead. At this point, the iterator object is exhausted and any further calls to its __next__() method just raise StopIteration again. Iterators are required to have an __iter__() method that returns the iterator object itself so every iterator is also iterable and may be used in most places where other iterables are accepted. One notable exception is code which attempts multiple iteration passes. A container object (such as a list) produces a fresh new iterator each time you pass it to the iter() function or use it in a for loop. Attempting this with an iterator will just return the same exhausted iterator object used in the previous iteration pass, making it appear like an empty container.

  • 可迭代的 Iterable对象。 可以使用isinstance(x, Iterable) 判断。
    集合数据类型: list tuple dict set str
    生成器 generator:生成器 & 生成器函数yield
>>> from collections.abc import Iterable
>>> isinstance([], Iterable)
True
>>> isinstance({}, Iterable)
True
>>> isinstance('abc', Iterable)
True
>>> isinstance((x for x in range(10)), Iterable)
True
>>> isinstance(100, Iterable)
False
  • 迭代器也可以通过isinstance(x, Iterator) 判断。
>>> from collections.abc import Iterator
>>> isinstance((x for x in range(10)), Iterator)
True
>>> isinstance([], Iterator)
False
>>> isinstance({}, Iterator)
False
>>> isinstance('abc', Iterator)
False
  • list dict之类可以迭代但是不是迭代器
  • 可作用于for循环的对象都是Iterable类型
  • 可作用于next()函数的对象都是Iterator类型,它们表示一个惰性计算的序列
  • 可以对listIterable但是不是Iterator的类型使用iter()函数,获得一个Iterator对象
>>> from collections.abc import Iterator
>>> isinstance(iter({}), Iterable)
True
>>> isinstance(iter({}), Iterator)
True
  • 总结
    一个实现了iter方法的对象是可迭代的,一个实现next方法并且是可迭代的对象是迭代器
    可以被next()函数调用并不断返回下一个值的对象称为迭代器Iterator

所以一个实现了iter方法和next方法的对象就是迭代器


Reference

[1] https://docs.python.org/3/glossary.html#term-generator

[2] python 生成器和迭代器有这篇就够了

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

推荐阅读更多精彩内容