python实用技巧：在列表，字典，集合中快速筛选数据

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)

源码下载

源码包下载