概 述

Python 列表推導(dǎo)式(list comprehension)利用其他列表創(chuàng)建新的列表，工作方式類似于for循環(huán)，使用列表推導(dǎo)式可以快速生成一個(gè)列表，或者根據(jù)某個(gè)列表生成滿足指定需求的列表。

單循環(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)+多條件

第一個(gè)字母+第二個(gè)字母均相同

[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]) 
<generator object <genexpr> 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<->arnold 
bernice<->bob 
bernice<->bernod 
clarice<->chris 

效率問題

男孩-女孩名字對(duì)的例子中，其實(shí)效率不是很高，因?yàn)槌绦驎?huì)檢測每個(gè)可能的配對(duì)。Python中有很多解決這個(gè)問題的方法，下面是其中一種比較高效的方法。

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()方法，是字典的一個(gè)方法，類似于get()方法，如果鍵不存在于字典中，將會(huì)添加鍵并將值設(shè)為默認(rèn)值。如果存在，則返回該字段的值。

語法：

dict.setdefault(key, default=None)