python中,要对列表、字典、集合进行数据筛选,最简单的方式就是用遍历,逐一对比,将符合条件的元素保存。这种方式虽然简单,但不够简洁优雅,以下用实例说明其他实现方式。
本文示例代码均用python3实现
列表、字典、集合解析
筛选列表数据
构建一个数值范围在-5至20的10个元素的列表,并将该列表中大于3的数据取出
构建列表
from random import randint
data = [randint(-5, 20) for _ in range(10)] # 表示循环了10次,每次循环都从-5至20之间取一个数值保存到data中
print(data)
用遍历的方式筛选数据
'''迭代'''
for x in data:
if x > 3:
result.append(x)
print(result)
使用filter函数
'''filter函数'''
result = list(filter(lambda x: x > 3, data)) # 使用匿名函数对列表data进行过滤
print(result)
使用列表解析
'''列表解析'''
result = [x for x in data if x > 3]
print(result)
筛选字典元素
假设一个班里有30个学生,python的考试成绩为0至100分之间,要将<60分的同学筛选出来。
针对此需求,需要先构建一个以学号为key,分数为value的字典
'''构建字典'''
student_score = {x: randint(0, 100) for x in range(1, 31)}
print(student_score)
使用字典解析
result = {k:v for k, v in student_score.items() if v < 60}
print(result)
集合解析
筛选一个集合中的偶数
构建集合
myset = {randint(5, 20) for _ in range(20)} # set集合中不能包含重复的数据,循环20次有可能获取到重复的数据,因此元素的个数可能小于20个
print(myset)
print(type(myset))
使用集合解析
result = {x for x in myset if x % 2 == 0}
print(result)
源码下载