2020-01-08一起来学习python中的生成器吧

什么是生成器

在学习生成器之前,首先我们要知道什么是生成器? 在python中一边循环一边计算的机制就是生成器(generator)

生成器作用

知道了什么是生成器之后,我们需要知道为什么要有生成器呢? 也就是说这种一边循环一边计算的机制有什么用处呢?

我们知道列表中所有的数据都存在内存中,如果数据很多的话将会特别消耗内存。先不说内存有没有这么大,假设我们仅仅需要访问前面几个元素,那后面元素所占用的空间岂不是都白费了吗。

所以这个时候如果列表中的元素能够按照某种算法自动推算出来呢,是不是就不必创建完整的列表,可以节省大量的内存空间了。所以如果又想节省内存,又需要用到很多数据就可以使用生成器。

创建生成器

通过类似列表生成式方式实现

创建生成器有好几种方法,最简单一种就是把一个列表生成式的[]改成(),这样就能直接创建一个生成器:

# 列表生成式list=[xforxinrange(10)]print(list)# [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]# 生成器generator=(xforxinrange(10))print(generator)# <generator object <genexpr> at 0x0000000002131A20>

我们可以直接通过print(list),将list中所有元素打印出来。但是print(generator)打印出来的却是一个对象,如果我们想要打印出生成器中的元素要怎么帮?有两种方式:

:通过next()方法一个一个元素打印:

generator=(xforxinrange(10))print(next(generator))# 0print(next(generator))# 1print(next(generator))# 2

每次调用一次next()方法就计算出下一个元素的值,直到超出最后一个元素时,则会抛出StopIteration错误。

这种方法只能一个一个元素打印,如果只需要前面几个元素,用这样方法刚好。如果需要打印很多元素,这个方法就太不方便了。这时候我们可以用第二种方法。

:使用for循环,因为生成器也是可迭代对象。

generator=(xforxinrange(5))forgingenerator:print(g)# 输出:01234

这样不仅可以一次性将所有元素打印出来,且通过for循环打印不需要担心会发生StopIteration异常。

这样看起来和列表生成式没差?但我们要注意列表生成式是直接将所有元素存在了内存空间中,这样占用了大量内存。而生成器并不是立即将结果写入内存, 而是保存的一种计算方式, 通过不断的计算, 可以获取到相应的位置的值,所以占用的内存仅仅是对计算对象的保存。

方法二:通过函数来实现

如果推算的算法比较复杂,用类似列表生成式的for循环无法实现的时候,还可以用函数来实现。只需要在函数中定义一个yield语句就行。

所以带有 yield 的函数不再是一个普通函数,而是一个生成器。yield相当于 return 返回一个值,并且记住这个返回的位置,下次迭代时,代码从yield的下一条语句开始执行。

defgen_yield():foriinrange(3):yieldig=gen_yield()print(next(g))# 0print(next(g))# 1print(next(g))# 2print(next(g))# 报错:StopIteration# 生成器函数每一次next()迭代,会返回当前yield的值,且在此处暂停,下一次next()迭代,# 从上一次的yield处开始,向下执行,且依然在下一次的yield中暂停# for循环迭代foriing:print(i)# 输出:012

上述中gen_yield()其实就是一个生成器函数,如果我们想要得到这个函数中的返回值,同样可以通过next()方法或者for循环迭代来得到。

参考:

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

推荐阅读更多精彩内容