Python 列表生成式 & 字典生成式

Python 列表生成式 & 字典生成式

經過生成式能夠更加簡潔地生成列表和字典

列表生成式

對比

直接生成數據後加入列表示例：

user_list = list()
for i in range(10):
    user_list.append(i ** 2)
print(user_list)

經過列表生成式直接生成示例：

user_list = [x ** 2 for x in range(10)]
print(user_list)

輸出結果相同：

[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

經過列表生成式的方法明顯簡潔許多

其餘示例

列表生成式中還可使用 if 語句
獲取原始列表中大於 5 的數示例：

original_list = [1, 2, 3, 4, 5, 6, 7, 8]
user_list = [x for x in original_list if x > 5]
print(user_list)

輸出結果：

[6, 7, 8]

字典生成式

字典生成式相似於列表生成式

對比

直接生成數據後加入列表示例：

user_dict = dict()
for i in range(10):
    user_dict[i] = i ** 2
print(user_dict)

經過字典生成式直接生成示例：

user_dict = {x: x ** 2 for x in range(10)}
print(user_dict)

輸出結果相同：

{0: 0, 1: 1, 2: 4, 3: 9, 4: 16, 5: 25, 6: 36, 7: 49, 8: 64, 9: 81}

其餘示例

字典生成式一樣可使用 if 語句
獲取原始字典中值爲奇數的鍵值對示例：

original_dict = {0: 0, 1: 1, 2: 4, 3: 9, 4: 16, 5: 25, 6: 36, 7: 49, 8: 64, 9: 81}
user_dict = {k: v for k, v in original_dict.items() if v % 2}
print(user_dict)