Python基础之推导式

Posted 2022-12-03

⼀、列表推导式

作⽤：⽤⼀个表达式创建⼀个有规律的列表或控制⼀个有规律列表。

列表推导式⼜叫列表⽣成式。

1、示例

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

2、for循环实现
list1 = []
for i in range(10):
 list1.append(i)
print(list1)

3、列表推导式实现
list1 = [i for i in range(10)]
print(list1)

2、带if的列表推导式

需求：创建0-10的偶数列表
⽅法⼀：range()步⻓实现
list1 = [i for i in range(0, 10, 2)]
print(list1)

⽅法⼆：if实现
list1 = [i for i in range(10) if i % 2 == 0]
print(list1)

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

需求：创建列表如下：
[(1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]
list1 = [(i, j) for i in range(1, 3) for j in range(3)]
print(list1)

⼆、字典推导式

list1 = [name, age, gender]
list2 = [Tom, 20, man]
如何快速合并为⼀个字典？
答：字典推导式
字典推导式作⽤：快速合并列表为字典或提取字典中⽬标数据。

1、示例

创建⼀个字典：字典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

2、将两个列表合并为⼀个字典

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

3、提取字典中⽬标数据

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

三、集合推导式

需求：创建⼀个集合，数据为下⽅列表的2次⽅。
list1 = [1, 1, 2]
set1 = i ** 2 for i in list1
print(set1) # 1, 4

注意：集合有数据去重功能。