Python生成器

1 生成器概念

生成器是一个特殊的迭代器(迭代器的抽象层级更高)
所以,生成器拥有迭代器的特性:1.惰性计算,节省内存。2.更够记录状态,并通过next()函数,访问下一个状态。3.具备可迭代特性。

2 生成器的的创建

(1)使用生成器表达式,示例代码如下:

# l = [i for i in range(1, 1000) if i % 2 ==0]
# 直接生成一个列表
l = (i for i in range(1, 1000) if i % 2 ==0)
# 直接生成一个生成器
print(next(l))
print(next(l))
print(l.__next__())
>>> 2
>>> 4
>>> 6

(2)使用生成器函数

函数中包含yield语句,这个函数执行结果就是“生成器”,示例代码如下:
yield,可以去阻断当前的函数运行,使用next()函数或next()方法,都会让函数继续执行,执行到下一个yield语句并把yield后的值当做状态值返回,函数到此处暂停。
特别说明:不使用next()或next()方法,函数不会执行,一行都不会执行,只会返回一个生成器。

def test():
  print("xxx")
  yield 1
  print("a")
  yield 2
  print("b")
  yield 3
  print("c")
  yield 4
  print("d")

g = test()
print(g) 
print(next(g))
print(next(g))
print(next(g))
>>> <generator object test at 0x000000002134d5>
>>> xxx
>>> 1
>>> a
>>> 2
>>> b
>>> 3

当生成器内的yield语句运行完之后,再一直使用next()或next()方法,就会报错StopIteration

也可以使用for in 来遍历,这种方法可以把最后一个yield下面的语句也运行出来。

def test():
  print("xxx")
  yield 1
  print("a")
  yield 2
  print("b")
  yield 3
  print("c")
g = test()

for i in g:
    print(i)
>>> xxx
>>> 1
>>> a
>>> 2
>>> b
>>> 3
>>> c

3 send()方法

(1)send()方法有一个参数,指的是上一次被挂起的yield语句返回值
(2)相比于.next(),send()可以额外的给yield传值
(3)注意第一次使用,t.send(None)
next()方法只是执行,不会传入值

def test():
    print("xxx")
    res1 = yield 1
    print(res1)
    res2 = yield 2
    print(res2)
g = test()

print(g.__next__())
print(g.__next__())
>>> xxx
>>> 1
>>> None # res1的值为None
>>> 2

send()方法给res1传入了值ooo,send()执行的位置在yield2处,但是它传入的值给到了上一次被挂起的yield1处。

def test():
    print("xxx")
    res1 = yield 1
    print(res1)
    res2 = yield 2
    print(res2)
g = test()

print(g.__next__())
print(g.send("ooo"))
>>> xxx
>>> 1
>>> ooo  # res1的值为ooo
>>> 2

直接执行send()传值操作会报错,因为它只给上一个yield挂起位置传值。

def test():
    print("xxx")
    res1 = yield 1
    print(res1)
    res2 = yield 2
    print(res2)
g = test()

print(g.send("ooo"))
>>> TypeError: can't send non-None value to a just-started generator

g.send(None) <=> g.next(),我们可以使用send(None),运行第一个yield,到第二个再使用send()传值就可以了。

def test():
    print("xxx")
    res1 = yield 1
    print(res1)
    res2 = yield 2
    print(res2)
g = test()

print(g.send(None))
print(g.send("ooo"))
>>> xxx
>>> 1
>>> ooo
>>> 2

4 close()方法关闭生成器

使用close()方法可以直接关闭生成器,再次调用生成器,会报错StopIteration

def test():
    yield 1
    print("a")
    yield 2
    print("b")
    yield 3
    print("c")
g = test()

print(g.__next__())
print(g.__next__())
g.close()
print(g.__next__())
>>> 1
>>> a
>>> 2
>>> StopIteration

5 注意事项:

如果碰到return,生成器会直接终止,抛出StopIteration异常提示

def test():
    yield 1
    print("a")
    yield 2
    print("b")
    return 10
    yield 3
    print("c")
g = test()

print(g.__next__())
print(g.__next__())
print(g.__next__())
>>> 1
>>> a
>>> 2
>>> 3
>>> StopIteration:10

生成器只会遍历一次,再次遍历需要重新创建,比如再次使用g = test()

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

推荐阅读更多精彩内容