零基礎入門學習Python(26):生成器
生成器(generator):在普通的函數中加上 yield ,使Python模仿協同程序的概念得以實現。
所謂的協同程序就是可以運行的獨立函數調用,函數可以暫停或者掛起,並在需要的時候從程序離開的地方繼續或重新開始。
>>> def myGen():
print('生成器被執行!')
yield 1 #對於生成器,出現yield,它會把後面的參數返回,然後暫停在yield這個位置,下一次執行就在這裏
yield 2
>>> myG = myGen()
>>> next(myG) #生成器是一個特殊的迭代器,是迭代器的一個實現
生成器被執行!
1
>>> next(myG)
2
>>> next(myG) #沒有值時會拋出StopIteration異常
Traceback (most recent call last):
File "<pyshell#8>", line 1, in <module>
next(myG)
StopIteration
>>> for i in myGen(): #for循環會自動調用next方法和探測StopIteration,是否結束
print(i)
生成器被執行!
1
2
下面通過舉例介紹Python中的各種推導式:
>>> a = [i for i in range(100) if not(i % 2) and i % 3] #列表推導式
>>> a #得到能被2整除且不能被3整除的100以內的數
[2, 4, 8, 10, 14, 16, 20, 22, 26, 28, 32, 34, 38, 40, 44, 46, 50, 52, 56, 58, 62, 64, 68, 70, 74, 76, 80, 82, 86, 88, 92, 94, 98]
>>> b = {i:i % 2 == 0 for i in range(10)} #字典推導式;此處,i % 2 == 0是條件表達式,==0時爲True,!=0時爲False
>>> b
{0: True, 1: False, 2: True, 3: False, 4: True, 5: False, 6: True, 7: False, 8: True, 9: False}
>>> c = {i for i in [1,1,2,3,4,5,5,6,7,8,3,2,1]} #集合推導式
>>> c #剔除重複的元素
{1, 2, 3, 4, 5, 6, 7, 8}
>>> d = "i for i in 'Hello World !'"
>>> d #直接打印出來
"i for i in 'Hello World !'"
>>> e = (i for i in range(10)) #用圓括號(),生成器推導式
>>> e
<generator object <genexpr> at 0x025E9350>
>>> next(e)
0
>>> next(e)
1
>>> next(e)
2
>>> for each in e:
print(each)
3
4
5
6
7
8
9
>>> sum((i for i in range(100) if i % 2)) #生成器推導式作爲函數參數
2500
>>> sum(i for i in range(100) if i % 2) #生成器推導式如果作爲函數參數,可以直接寫,不用加()
2500