Python 列表推導(dǎo)式與生成器
推導(dǎo)式
會(huì)直接生成一個(gè)相應(yīng)的對(duì)象,也叫解析式
列表推導(dǎo)式
list_i = [i for i in range(0,25)]
print(list_i)
print(type(list_i))
#[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24]
#<class 'list'>
可以很明顯的看出列表推導(dǎo)式就是快速的生成了一個(gè)列表
當(dāng)然,可以增加篩選條件
list_i = [i for i in range(0,25) if i % 2 != 0]#篩選出0到24的所有奇數(shù)
print(list_i)#[1, 3, 5, 7, 9, 11, 13, 15, 17, 19, 21, 23]
其他的推導(dǎo)式也相同
字典推導(dǎo)式
x = ['a','b','c']
y = [1,2,3]
dict_i = {k: v for k,v in zip(x,y)}
print(dict_i) #{'a': 1, 'b': 2, 'c': 3}
#增加篩選條件,鍵不為a并且值大于2
dict_i = {k: v for k,v in zip(x,y) if k != 'a'and v > 2}
print(dict_i) #{'c': 3}
集合推導(dǎo)式
set_i = {i*i for i in [1,1,2,2,4,5,7]}
print(set_i)#{1, 4, 16, 49, 25} 已去重
print(type(set_i))#<class 'set'>
優(yōu)缺點(diǎn)
推導(dǎo)式可以快速的生成列表、字典和集合,方便快捷。
但是會(huì)一次性占用很大空間
生成器
解決推導(dǎo)式空間占有的方法就是生成器了,生成器不會(huì)一次性的占有一塊存儲(chǔ)空間,而是通過(guò)yield的方式一個(gè)一個(gè)的生成。除此之外生成器還有保存狀態(tài)的特性。
調(diào)用了yield的函數(shù)叫做生成器函數(shù),它返回一個(gè)迭代器。
使用案例
def fibonacci(num):#斐波那契數(shù)列生成器
a,b,counter = 0,1,0
while True:
if (counter > num):
return
yield a
a,b = b, a+b
counter += 1
f = fibonacci(10)
print(f)#<generator object fibonacci at 0x10759fa40> #返回一個(gè)生成器對(duì)象
print(next(f))
print(next(f))
print(next(f))
print(next(f))
'''
#和迭代器一樣,通過(guò)next的方式輸出下一個(gè)元素
0
1
1
2
3
'''
保存狀態(tài)與StopIteration
注意,接著以上代碼,加上
while True:
print(next(f))
得到的結(jié)果是
'''
5
8
13
21
34
55
---------------------------------------------------------------------------
StopIteration Traceback (most recent call last)
~/wencyyyyyy/iterator_and_generator.py in
1 while True:
----> 2 print(next(f))
StopIteration:
'''
這里并非從斐波那契數(shù)列的0開(kāi)始,而是接著上面已經(jīng)輸出過(guò)的3,接著往下5、8、13、21、34、55
是因?yàn)樵谡{(diào)用生成器運(yùn)行的過(guò)程中,每次遇到 yield 時(shí)函數(shù)會(huì)暫停并保存當(dāng)前所有的運(yùn)行信息,返回 yield 的值, 并在下一次執(zhí)行 next() 方法時(shí)從當(dāng)前位置繼續(xù)運(yùn)行,這就是生成器的==保存狀態(tài)==。
保存狀態(tài)
這里在原代碼中加入一行標(biāo)志
def fibonacci(num):
a,b,counter = 0,1,0
while True:
if (counter > num):
return
yield a
print('這是第{}次'.format(counter))
a,b = b, a+b
counter += 1
f = fibonacci(10)
print(f)
print(next(f))
'''
<generator object fibonacci at 0x1076e7830>
0
'''
這里并沒(méi)有輸出,這是第counter次。說(shuō)明函數(shù)確實(shí)執(zhí)行到y(tǒng)ield就暫停了。那么下次next(f)時(shí)有沒(méi)有繼續(xù)從上次結(jié)束的地方運(yùn)行呢?
print(next(f))
'''
這是第0次
1
'''
很明顯,確實(shí)是從上次結(jié)束的的地方繼續(xù)運(yùn)行的。
StopIteration
此外,出現(xiàn)了一個(gè)異常,StopIteration 。這里沒(méi)什么好解釋的,因?yàn)樯善鞣祷氐氖且粋€(gè)迭代器。迭代完成,斐波那契數(shù)列中數(shù)字超過(guò)10個(gè)時(shí),就代表迭代已經(jīng)完成。
聯(lián)系方式
如果您對(duì)本片博文有任何意見(jiàn)或者建議,歡迎您聯(lián)系我