Python推导式

推导式

一、什么是推导式

通过我们直接给定值、或者通过其他序列转换之外，我们还可以通过推导式进行创建一个列表、字典或集合。

⽤⼀个表达式创建⼀个有规律的容器。

二、列表推导式

list = [表达式]

普通实现：

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

1. 推导式实现：

list1 = [i for i in range(10)]
print(list1) # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

2. 带if的推导式

list2 = [i for i in range(10) if i % 2 == 0]
print(list2) # [0, 2, 4, 6, 8]

3. 多个for循环实现的推导式

list3 = [(i, j) for i in range(1, 3) for j in range(2)]
print(list3) # [(1, 0), (1, 1), (2, 0), (2, 1)]

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

三、字典推导式

• 创建一个字典： 字典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) # {'name': 'Tom', 'age': 20, 'gender': 'man'}

带if的推导式

counts = {'MBP': 268, 'HP': 125, 'DELL': 201, 'Lenovo': 199}
# 需求：提取上述电脑数量⼤于等于200的字典数据
count1 = {key: value for key, value in counts.items() if value > 200}
print(count1) # {'MBP': 268, 'DELL': 201}

四、集合推导式

和列表用法基本一致

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