Python 的列表推導(dǎo)式、字典推導(dǎo)式 列表推導(dǎo)式

列表解析是用一行代碼實(shí)現(xiàn)根據(jù)一個(gè)列表生成另一個(gè)列表的寫法。

基本寫法

要實(shí)現(xiàn)從一個(gè)列表中獲取滿足條件的某些元素組成新的列表：

[x for x in [1,2,3,4] if x % 2 == 0]

這行代碼輸出是 [2, 4]。如果沒有 if 的話就是直接復(fù)制整個(gè)列表。

其他用法

修改新列表的元素：

[x/2 for x in [1,2,3,4]]

將單個(gè)元素的列表重復(fù)此元素，擴(kuò)展為指定長度的列表：

[x for i in range(100)]

提取嵌套列表中元素的第一項(xiàng)：

[x[0] for x in [[1,2], [3,4], [5,6]]]

嵌套循環(huán)的寫法：

[n for row in [[1,2], [3,4], [5,6]] for n in row]

生成嵌套列表：

[(x,x+10) for x in [1,2]

多個(gè)參數(shù)：
[(x,y) for x,y in {1:2, 4:5}.items() if x % 2 == 0]

與條件表達(dá)式結(jié)合：
[x if x % 3 == 0 else 1 for x in range(10)]

[(x,y) if x % 2 == 0 else (y, 3) for x,y in {1:2, 4:5}.items() ]

字典推導(dǎo)式

與列表推導(dǎo)式類似，字典也有相應(yīng)的推導(dǎo)式寫法。

基本寫法

{x:y for y,x in {'1':'2', '3':'4'}.items() if '1' in y}

參考資料：

《輕松學(xué)會(huì)Python列表解析式》
《字典推導(dǎo)式》