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

目录

列表推导式

带if的列表推导式

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

字典推导式

集合推导式

---

列表推导式

作用：用一个表达式创建一个有规律的列表或控制一个有规律的列表（化简代码）

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

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

带if的列表推导式

list = [i for i in range(10) if i % 2==0]
print(list)
输出：[0, 2, 4, 6, 8]

多个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)]

字典推导式

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

dict1 = { i : i**2 for i in range(1,5)}
print(dict1)
输出：{1: 1, 2: 4, 3: 9, 4: 16}

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

list1 = ['name','age','gender','id']
list2 = ['TOM','20','男']
dict1 = {list1[i] : list2[i] for i in range(min(len(list1),len(list2)))}
print(dict1)
输出：{'name': 'TOM', 'age': '20', 'gender': '男'}

总结：

①如果两个列表数据个数相同，len统计任何一个列表的长度都可以

②如果两个列表数据个数都不同，len统计数据多的列表数据个数会报错；len统计数据少的列表个数正常运行

#提取字典中的目标数据

dict = {'MBP':268,'HP':125,'DELL':201,'LENOVO':199,'ACER':99}
dict1 = {key : value for key, value in dict.items() if value >= 200}
print(dict1)
输出：{'MBP': 268, 'DELL': 201}

集合推导式

list1 = {1,1,2}
set1 = {i**2 for i in list1}
print(set1)
输出：{1, 4}