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)