python中推导式

目标

• 列表推导式
• 字典推导式
• 集合推导式

一、列表推导式（列表生成式）

作用：用一个表达式来创建或控制一个有规律的列表

1.1快速体验

需求：创建一个0-10的列表

• while实现

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

• for实现

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

• 列表推导式实现

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

结果：

 由此可以看出，列表推导式是十分简洁的。

1.2带if的列表推导式

需求：创建一个0-10的偶数列表

• 方法1：range()步长实现

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

• 方法2：if实现

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

结果：

 1.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)

效果和两个for循环嵌套一样，但推导式更简洁

二、字典推导式

2.1快速体验

1.创建一个字典：字典的key是1-5数字，value是这个数字的平方

dict1 = {i: i**2 for i in range(1, 6)}
print(dict1)

结果：

2. 两个列表合并为字典

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

结果：

3.提取字典中的目标数据（items()：可以提取出字典的键和值）

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)

结果：

 三、集合推导式

需求：创建一个集合，数据为下方列表的平方

list1 = [1, 1, 2]

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

结果：（集合有去重的作用）

 四、总结

• 推导式的作用：简化代码
• 推导式的写法：

# 列表推导式
[xx for xx in range()]

# 字典推导式
{xx1: xx2 for ... in ...}

# 集合推导式
{xx for xx in ...}