Python学习笔记（十一）：推导式的使用

1列表推导式

⽤⼀个表达式创建⼀个有规律的列表或控制⼀个有规律列表。列表推导式⼜叫列表⽣成式。看这个描述一开始着实难懂，但实际上只是用了for循环来创建列表，只不过使用推导式比for循环更加简洁，速度也更快一点。

1.1 一般的推导式

假设我们需要构造一个存储数字0到9的列表，我们可以通过while循环或者for循环来实现，只不过代码量多一些，如下。

• while循环实现

# 1. 准备⼀个空列表
list1 = []
# 2. 书写循环，依次追加数字到空列表list1中
i = 0
while i < 10:
 list1.append(i)
 i += 1
print(list1)

• for循环实现

list1 = []
for i in range(10):
 list1.append(i)
print(list1)

但如果我们使用推导式，代码就可以变得更加简洁。

• 推导式

list1 = [i for i in range(10)]
print(list1)

从这里看，推导式只不过在for循环前加上了循环变量i，并将每个i返回了出来。

1.2 带if的推导式

我们可以在推导式的for循环后加入if判断语句，用于选择哪些i将被输出。例如，仅当i是偶数时才被选中。

list1 = [i for i in range(10) if i % 2 == 0]
print(list1)

当然这也可用通过控制for循环的步长来实现。

1.3 多个for循环实现列表推导式

推导式也可以进行for循环的嵌套，和一般的for循环嵌套也没有什么区别。

list1 = [(i, j) for i in range(1, 3) for j in range(3)]
print(list1)

[(1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]

字典推导式

也可以通过推导式来构建字典，只不过for循环前需要有两个变量，分别表示key和value。

• 创建⼀个字典：字典key是1-5数字，value是这个数字的2次⽅。

dict1 = {i: i**2 for i in range(1, 5)}
print(dict1) # {1: 1, 2: 4, 3: 9, 4: 16}

• 将两个列表合并为一个字典

list1 = ['name', 'age', 'gender']
list2 = ['Tom', 20, 'man']
dict1 = {list1[i]: list2[i] for i in range(len(list1))}
print(dict1)

集合推导式

完全和上面一样，唯一要注意的是集合有数据去重功能。

• 创建⼀个集合，数据为下⽅列表的2次⽅。

list1 = [1, 1, 2]

list1 = [1, 1, 2]
set1 = {i ** 2 for i in list1}
print(set1) # {1, 4}