五分钟了解 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)

此时使用大括号就变成了一个集合推导式，用中括号就是一个列表推导式。