五分鐘瞭解 Python 列表推導式，集合推導式，字典推導式

Python中有一種特別的表達式，就是推導式，包括列表推導式，集合推導式，字典推導式。可以通過輸入一個序列，通常是for循環，加上過濾條件，生成一個新的序列。典型結構如下圖所示：

輸出結果：

可以看出所有小於10的偶數的平方生成了一個新的列表，當然我們也可以通過多個條件來過濾，有兩種格式如下：

n_list = [x ** 2 for x in range(100) if x % 2 == 0 if x % 5 ==0]
n_list1 = [x ** 2 for x in range(100) if x % 2 == 0 and x % 5 ==0]
print(n_list)
print(n_list1)

上面的列表推導式實際等效於下面的代碼：

n_list = []
for x in range(100):
    if x % 2 == 0 and x % 5 == 0:
        n_list.append(x**2)
print(n_list)

三種形式輸出都是相同的。

---

對於集合推導式而言，其形式和用法與列表推導式基本一致，只需要將中括號改爲大括號即可

n_list1 = {x ** 2 for x in range(100) if x % 2 == 0 and x % 5 ==0}
print(n_list1)

列表變成集合，這裏需要注意的是，集合沒有重複元素但是列表可以有重複的元素，因此在最後集合推導式中會將重複的元素捨棄。同理，上面的代碼等效於：

n_list = set()
for x in range(100):
    if x % 2 == 0 and x % 5 == 0:
        n_list.add(x**2)
print(n_list)

---

最值得注意的是字典推導式的用法，因爲字典中有鍵和鍵值兩個元素對應存放，要稍微複雜些。

n_dic = {1:'one',2:'two',3:'three'}
print(n_dic)
out_dic = {k:v for k,v in n_dic.items() if k<3}
print(out_dic)

我們之前講過，要在一個序列中取值，但是字典不是一個序列，需要調用items方法轉換成序列。

字典還有其他比較靈活的使用方法，可以只取鍵或者鍵值，比如：

n_dic = {1:'one',2:'two',3:'three'}
print(n_dic)
out_dic = {k for k,v in n_dic.items() if k<3}
print(out_dic)

此時使用大括號就變成了一個集合推導式，用中括號就是一個列表推導式。