python推导式

内容

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

列表推导式

作用：用一个表达式创建一个有规律的列表或控制一个有规律的列表
列表推导式又叫列表生成式

创建有规律的列表

• while循环实现
• for循环实现

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

• 列表推导式简化代码：创建或者空控制有规律的列表
• 带if的列表推导式
• 多个if循环实现列表推导式

# 列表推导式
list1 = [i for i in range(10)]
print(list1)
# 带if的列表推导式
list2 = [i for i in range(10) if i % 2 == 0]
print(list2)
# 多个for循环实现列表推导式
list3 = [(i, j) for i in range(1, 3) for j in range(3)]
print(list3)
'''
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
[0, 2, 4, 6, 8]
[(1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]
'''

字典推导式

字典推导式作用：快速合并列表为字典或提取字典中的目标数据

• 创建一个字典，key是1-5数字，value是这个数字的2次方

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

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

# 将两个列表合并为一个字典
list1 = ['name', 'age', 'gender']
list2 = ['Tom', 20, 'man']
dict2 = {list1[i]: list2[i] for i in range(len(list1))}
print(dict2)

总结：如果两个数据相同，len统计任何一个列表重读都可以；如果两个数据数量不同，统计数据多的报错，统计数据少的不报错

• 提取字典中的目标数据

集合推导式

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