如何理解Python字典推导式和集合推导式

288 阅读2分钟

 

 

思考:如果有以下两个列表,如何快速合并为一个字典?


list1 = ['name', 'age', 'sex']
list2 = ['Python自学网', '30', '女']

答: 使用字典推导式

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

 

一、 字典推导式 快速体验:

1、创建一个字典,字典key是1-5数字,value是这个数字的2次方。

 

代码体验:

# dict1 = {k:v for i in range(1, 5)}
dict1 = {i: i**2 for i in range(1, 5)}
print(dict1)
# {1: 1, 2: 4, 3: 9, 4: 16}

 

2、将2个列表合并为一个字典

代码体验:

 

# 2个列表数据个数一样
list1 = ['name', 'age', 'sex']
list2 = ['Python自学网', '30', '女']

dict1 = {list1[i]:list2[i] for i in range(len(list1))}
print(dict1)
# {'name': 'Python自学网', 'age': '30', 'sex': '女'}

# 2个列表数据个数不一样
list1 = ['name', 'age', 'sex', 'id']
list2 = ['Python自学网', '30', '女']

dict1 = {list1[i]:list2[i] for i in range(len(list2))}
print(dict1)
# {'name': 'Python自学网', 'age': '30', 'sex': '女'}

   

总结:

1、如果两个列表数据个数相同,len()统计任何一个列表的长度都可以

2、如果两个列表数据个数不相同,len()统计数据多的列表数据个数会报错,len()统计数据少的列表数据个数不会报错

 

3、提取字典中目标数据

 

需求:提取上述电脑数量大于等于200的字典数据

 

代码体验:

 

counts = {'HP': 189, 'DELL': 220, 'Lenovo': 180, 'MBP': 261, 'acer': 65}

# 获取所有键值对数据, 判断v值大于等于200 ---》 返回字典
# print(counts.items())

dict1 = {key: value for key, value in counts.items() if value >= 200}\
print(dict1)
# {'DELL': 220, 'MBP': 261}

 

字典推导式私下多练练,工作场景中用到的比较多。

 

二、集合 推导式 快速体验:

集合推导式没有字典推导式在工作场景中用到的那么频繁。

 

需求:创建一个集合,数据为下方列表的2次方。

list1 = [2, 2, 4]

 

 

代码如下:

 

list1 = [2, 2, 4]

set1 = {i ** 2 for i in list1}
print(set1)  # {16, 4}

 

注意:  集合有数据去重功能,所以这个集合数据只有2个数据分别是16,4

 

 

文章借鉴来源:Python自学网