Python奇技淫巧—[1]—在列表、字典、集合中根据条件筛选数据

Python奇淫巧技——在列表、字典、集合中根据条件筛选数据

通用做法:迭代

以列表为例:

筛选出下列数字大于等于0的数

data = [2, 7, -4, -1, 3, 0, 8]

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

print(res)
运行结果:
[2, 7, 3, 0, 8]
奇淫巧技——列表筛选
  • 使用filter函数

随机生成一组正负数皆有的数,筛选出大于等于0的数

flilter(function or None, iterable)

from random import randint


data = [randint(-10,10) for _ in range(10)]

# filter 接收两个参数,第一个参数为过滤条件,第二个参数为过滤范围
f = filter(lambda x: x>=0, data)
for j in f:
    print(j)
  • 使用列表解析(列表推导式)

随机生成一组正负数皆有的数,筛选出大于等于0的数

from random import randint


data = [randint(-10,10) for _ in range(10)]
f = [i for i in data if i >= 0]
print(f)

使用timeit测试两种方法的运行速度得到:使用列表解析的速度更快,但二者皆比通用做法迭代要快

奇淫巧技——字典筛选

假设有20个学生,学生分数在60-100之间,筛选出成绩在90分以上的学生

import random


dict = {i: random.randint(60, 100) for i in range(1, 21)}
# print(dict)
# 根据值过滤除分数高于90的人
res = {k: v for k, v in dict.items() if v > 90}
print(res)
奇淫巧技——集合筛选

10个随机的正负数,筛选出可以和2整除的数字

import random


data = set(random.randint(-10,10) for _ in range(10))
# print(data)
# 筛选集合中能被2整除的数
res = {i for i in data if i % 2 ==0}
print(res)

Leave a Comment