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)