Python实用技巧之列表、字典、集合筛选数据

以下所有列表、字典、集合都是通过random随机生成的：

import random
# 生成一个-10到10的10个随机整数列表
random_list = [random.randint(-10, 10) for _ in range(10)]

列表

1、过滤列表中的正数

使用for表达式

# 使用for表达式 内部使用if判断
r = [num for num in random_list if num >= 0]

使用filter高阶函数

# 这里使用lambda表达式
# filter函数返回的是一个迭代器对象，需要使用list内置方法转化
r = list(filter(lambda x: x >= 0, random_list))

字典

1、过滤字典中的正数值

# 生成一个学生字典，key为student从1到10，value为 -10到10的随机数
random_dict = {'student_%s' % (i+1) : random.randint(-10, 10) for i in range(10)}
# {'student_1': 1, 'student_2': 9, 'student_3': -3, 'student_4': 8, 'student_5': 9, 'student_6': -6, 'student_7': -2, 'student_8': 8, 'student_9': 0, 'student_10': -5}

使用for表达式

r = {key: value for key, value in random_dict.items() if value >= 0}

使用filter高阶函数

'''
小知识点：xxx.items() 输出的是一个由键值组成的元祖
'''
r1 = dict(filter(lambda x: x[1] >= 0, random_dict.items()))

set集合

过滤集合中的正数

# 随机生成一个-10到10的10个数的集合
random_set = {random.randint(-10, 10) for _ in range(10)}

for表达式：

r = {i for i in random_set if i >= 0}