欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Python 列表推导式,集合推导式,字典推导式

程序员文章站 2024-01-25 18:21:34
...

Python中有一种特别的表达式,就是推导式,包括列表推导式,集合推导式,字典推导式。可以通过输入一个序列,通常是for循环,加上过滤条件,生成一个新的序列。典型结构如下图所示:
Python 列表推导式,集合推导式,字典推导式
输出结果:
Python 列表推导式,集合推导式,字典推导式
可以看出所有小于10的偶数的平方生成了一个新的列表,当然我们也可以通过多个条件来过滤,有两种格式如下:

n_list = [x ** 2 for x in range(100) if x % 2 == 0 if x % 5 ==0]
n_list1 = [x ** 2 for x in range(100) if x % 2 == 0 and x % 5 ==0]
print(n_list)
print(n_list1)

Python 列表推导式,集合推导式,字典推导式
上面的列表推导式实际等效于下面的代码:

n_list = []
for x in range(100):
    if x % 2 == 0 and x % 5 == 0:
        n_list.append(x**2)
print(n_list)

Python 列表推导式,集合推导式,字典推导式
三种形式输出都是相同的。


对于集合推导式而言,其形式和用法与列表推导式基本一致,只需要将中括号改为大括号即可

n_list1 = {x ** 2 for x in range(100) if x % 2 == 0 and x % 5 ==0}
print(n_list1)

Python 列表推导式,集合推导式,字典推导式
列表变成集合,这里需要注意的是,集合没有重复元素但是列表可以有重复的元素,因此在最后集合推导式中会将重复的元素舍弃。同理,上面的代码等效于:

n_list = set()
for x in range(100):
    if x % 2 == 0 and x % 5 == 0:
        n_list.add(x**2)
print(n_list)

Python 列表推导式,集合推导式,字典推导式


最值得注意的是字典推导式的用法,因为字典中有键和键值两个元素对应存放,要稍微复杂些。

n_dic = {1:'one',2:'two',3:'three'}
print(n_dic)
out_dic = {k:v for k,v in n_dic.items() if k<3}
print(out_dic)

我们之前讲过,要在一个序列中取值,但是字典不是一个序列,需要调用items方法转换成序列。
Python 列表推导式,集合推导式,字典推导式
字典还有其他比较灵活的使用方法,可以只取键或者键值,比如:

n_dic = {1:'one',2:'two',3:'three'}
print(n_dic)
out_dic = {k for k,v in n_dic.items() if k<3}
print(out_dic)

Python 列表推导式,集合推导式,字典推导式
此时使用大括号就变成了一个集合推导式,用中括号就是一个列表推导式。

相关标签: Python笔记