推导式,又叫生成式,是一种更有风格的python代码写法,化简代码。包括: 1、列表推导式 2、字典推导式 3、集合推导式
一、列表推导式 作用:用一个表达式创建一个有规律的列表或控制一个有规律列表。 列表推导式又叫列表生成式。
需求:创建一个0-10的列表。用三种方法实现,对比出推导式如何化简代码。
''' 1.1 创建空列表 1.2 循环将有规律的数据写入到列表 ''' # while 实现 list1 = [] i = 0 while i < 10: list1.append(i) i += 1 print(list1) # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] # for 实现 list1 = [] for i in range(10): list1.append(i) print(list1) # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] # 列表推导式实现 list1 = [i for i in range(10)] print(list1) # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]1.2 带if的列表推导式
# 需求:创建0-10的偶数列表 # 1. 简单列表推导式range()步长 list1 = [i for i in range(0, 10, 2)] print(list1) # [0, 2, 4, 6, 8] # 2. for循环加if创建有规律的列表 list2 = [] for i in range(10): if i % 2 == 0: list2.append(i) print(list2) # [0, 2, 4, 6, 8] # 3. 把for循环配合if的代码 改写 带if的列表推导式 list3 = [i for i in range(10) if i % 2 == 0] print(list3) # [0, 2, 4, 6, 8]1.3 多个for实现列表推导式
# [(1,0),(1,1),(1,2),(2,0),(2,1),(2,2)] # 数据1: 1 和 2, range(1,3) # 数据2: 0 1 2 , range(3) list1 = [] for i in range(1, 3): for j in range(3): # 列表里面追加元组:循环前准备一个空列表,然后这里追加元组数据到列表 list1.append((i, j)) print(list1) # [(1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)] # 多个for 实现列表推导式 list2 = [(i, j) for i in range(1, 3) for j in range(3)] print(list2) # [(1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)] # 多for的列表推导式等同于for循环嵌套二、字典推导式 作用:快速合并列表为字典或提取字典中目标数据。
# 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) # {'name': 'Tom', 'age': 20, 'gender': 'man'} list1 = ['name', 'age', 'gender', 'id'] list2 = ['Tom', 20, 'man'] dict1 = {list1[i]: list2[i] for i in range(len(list1))} print(dict1) # IndexError: list index out of range list1 = ['name', 'age', 'gender', 'id'] list2 = ['Tom', 20, 'man'] dict1 = {list1[i]: list2[i] for i in range(len(list2))} print(dict1) # {'name': 'Tom', 'age': 20, 'gender': 'man'} # 总结: # 1. 如果两个列表数据个数相同,len统计任何一个列表的长度都可以 # 2. 如果两个列表数据个数不同,len统计数据多的列表数据个数会报错,len统计数据少的列表数据个数不会报错 # 3. 提取字典中目标数据 counts = {'MBP': 268, 'HP': 125, 'DELL': 201, 'Lenovo': 199, 'acer': 99} # 需求:提取上述电脑数量大于等于200的字典数据 # 获取所有键值对数据,判断v值大于等于200,返回字典 print(counts.items()) # dict_items([('MBP', 268), ('HP', 125), ('DELL', 201), ('Lenovo', 199), ('acer', 99)]) 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} # 集合有去重功能,所以这个集合数据只有两个数据分别是1,4总结:
# 列表推导式 [xx for xx in range()] # 字典推导式 {xx1: xx2 for ... in ...} # 集合推导式 {xx for xx in ...}