python 如何在列表、字典、集合中根据条件筛选数据--实战经验

风陵苑主2024-05-08 11:32

一、列表

需求：过滤掉列表中的负数数据

from random import randint
#生成数据
data = [randint(-10,10) for _ in range(10)]

复制代码

[-6, -10, -6, -9, -7, -7, 2, -9, 9, 2]

第一种，传统的方法

res=[]
for x in data:
    if x>=0:
        res.append(x)

第二种，filter方法

list(filter(lambda x:x>=0,data))
#[2, 9, 2]

第三种，列表解析

[x for x in data if x>=0]
#[2, 9, 2]

总结：无论是运行速度还是代码容易理解程度，列表解析最好用。

二、字典，筛选出字典里超过90分的数据

随机生成数据：

d = {f'{x}号':randint(60,100) for x in range(1,21)}
{
'1号': 92,
 '2号': 85,
 '3号': 85,
 '4号': 68,
 '5号': 86,
 '6号': 81,
 '7号': 70,
 '8号': 65,
 '9号': 86,
 '10号': 85,
 '11号': 86,
 '12号': 60,
 '13号': 94,
 '14号': 70,
 '15号': 89,
 '16号': 69,
 '17号': 71,
 '18号': 83,
 '19号': 60,
 '20号': 74
}

字典列表解析

{k:v for k,v in d.items() if v>90}
#{'1号': 92, '13号': 94}

三、集合解析

复制代码

{-10, -9, -7, -6, 2, 9}

要求过滤能被3整除的集合

{x for x in s if x%3==0}