列表推導式
-
列表推導式使用非常簡潔的方式來快速生成滿足特定需求的列表，代碼具有非常強的可讀性。
-
語法形式為：
[expression for expr1 in sequence1 if condition1
for expr2 in sequence2 if condition2
for expr3 in sequence3 if condition3
...
for exprN in sequenceN if conditionN]
-
列表推導式在邏輯上等價於一個循環語句，只是形式上更加簡潔。
例如：
aList = [x*x for x in range(10)]
相當於
>>> aList = []
>>> for x in range(10):
aList.append(x*x)
-
實現嵌套列表的平鋪
>>> vec = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
[num for elem in vec for num in elem]
[1, 2, 3, 4, 5, 6, 7, 8, 9]
在這個列表推導式中有2個循環，其中第一個循環可以看作是外循環，執行的慢；而第二個循環可以看作是內循環，執行的快。上面代碼的執行過程等價於下面的寫法：
>>> vec = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
>>> result = []
>>> for elem in vec:
for num in elem:
result.append(num)
>>> result
[1, 2, 3, 4, 5, 6, 7, 8, 9]
-
過濾不符合條件的元素
在列表推導式中可以使用if子句對列表中的元素進行篩選，只在結果列表中保留符合條件的元素。下面的代碼可以列出當前文件夾下所有Python源文件：
>>> import os
>>> [filename for filename in os.listdir('.') if filename.endswith(('.py', '.pyw'))]
下面的代碼用於從列表中選擇符合條件的元素組成新的列表：
>>> aList = [-1, -4, 6, 7.5, -2.3, 9, -11]
>>> [i for i in aList if i>0] #所有大於0的數字
[6, 7.5, 9]
-
同時遍歷多個列表或可迭代對象
[(x, y) for x in [1, 2, 3] for y in [3, 1, 4] if x != y]
[(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]
>>> [(x, y) for x in [1, 2, 3] if x==1 for y in [3, 1, 4] if y!=x]
[(1, 3), (1, 4)]
對於包含多個循環的列表推導式，一定要清楚多個循環的執行順序或“嵌套關系”。例如，上面第一個列表推導式等價於
>>> result = []
>>> for x in [1, 2, 3]:
for y in [3, 1, 4]:
if x != y:
result.append((x,y))
>>> result
[(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]
-
列表推導式中可以使用函數或復雜表達式
def f(v):
if v%2 == 0:
v = v**2
else:
v = v+1
return v
>>> [f(v) for v in [2, 3, 4, -1] if v>0]
[4, 4, 16]
>>> [v**2 if v%2 == 0 else v+1 for v in [2, 3, 4, -1] if v>0]
[4, 4, 16]
>>> x = list(range(10))
>>> [item>5 for item in x]
[False, False, False, False, False, False, True, True, True, True]