【Python基礎(chǔ)】Python輕量級循環(huán)-列表推導(dǎo)式
概 ?述
Python 列表推導(dǎo)式（list comprehension）利用其他列表創(chuàng)建新的列表，工作方式類似于for循環(huán)，使用列表推導(dǎo)式可以快速生成一個列表，或者根據(jù)某個列表生成滿足指定需求的列表。
單循環(huán)
[i for i in range(10)][0,?1,?2,?3,?4,?5,?6,?7,?8,?9]
單循環(huán)+條件
0-20中所有能被3整除的數(shù)是多少
[i for i in range(20) if i%3==0][0, 3, 6, 9, 12, 15, 18]
?
多循環(huán)-2次
'ABC'和'EGF'所有可能的兩兩組合
[i+j for i in 'ABC' for j in 'EFG']['AE',?'AF',?'AG',?'BE',?'BF',?'BG',?'CE',?'CF',?'CG']
多循環(huán)-3次
'ABC'、'EFG'、‘HIJ’所有可能的三三組合
[i+j+u for i in 'ABC' for j in 'EFG' for u in 'HIJ']['AEH', 'AEI', 'AEJ', 'AFH', 'AFI', 'AFJ', 'AGH', 'AGI', 'AGJ','BEH', 'BEI', 'BEJ', 'BFH', 'BFI', 'BFJ', 'BGH', 'BGI', 'BGJ',?'CEH',?'CEI',?'CEJ',?'CFH',?'CFI',?'CFJ',?'CGH',?'CGI',?'CGJ']
多循環(huán)+單條件
首字母相同的男孩 女孩組合
girls = ['alice','bernice','clarice']boys = ['chris','arnold','bob','bernod'][i+'<->'+j for i in girls for j in boys if i[0]==j[0]]['alice<->arnold', 'bernice<->bob', 'bernice<->bernod','clarice<->chris']
?
多循環(huán)+多條件
第一個字母+第二個字母均相同
[i+'<->'+j for i in girls for j in boys if (i[0]==j[0] and i[1]==j[1])]?['bernice<->bernod']
元組循環(huán)
列表可以直接循環(huán)，改成元組也是可以的，但是直接生產(chǎn)了迭代器
(i+'<->'+j for i in girls for j in boys if i[0]==j[0])at 0x0000015413B781C8> tuples = (i+'<->'+j for i in girls for j in boys if i[0]==j[0])for t in tuples:print(t)alice<->arnoldbernice<->bobbernice<->bernodclarice<->chris
效率問題
男孩-女孩名字對的例子中，其實效率不是很高，因為程序會檢測每個可能的配對。Python中有很多解決這個問題的方法，下面是其中一種比較高效的方法。
letterGirls = {}for girl in girls:letterGirls.setdefault(girl[0],[]).append(girl)print([i+'<->'+j for j in boys for i in letterGirls[j[0]]])['clarice<->chris', 'alice<->arnold', 'bernice<->bob','bernice<->bernod']
看看我們構(gòu)建的字典長啥樣
letterGirls{'a': ['alice'], 'b': ['bernice'], 'c': ['clarice']}
注意：setdefault()方法，是字典的一個方法，類似于get()方法，如果鍵不存在于字典中，將會添加鍵并將值設(shè)為默認(rèn)值。如果存在，則返回該字段的值。語法：dict.setdefault(key, default=None)
往期精彩回顧
本站知識星球“黃博的機器學(xué)習(xí)圈子”（92416895）
本站qq群704220115。
加入微信群請掃碼：
評論
圖片
表情