python 学习笔记(一)：在列表、字典、集合中根据条件筛选数据

一、在列表中筛选数据

在列表中筛选出大于等于零的数据，一般通用的用法代码如下：

data = [3, -9, 0, 1, -6, 3, -2, 8, -6] #要筛选的原始数据列表
result = [] #存放筛选结果的列表
for x in data: #依次迭代循环每个元素
    if x >= 0: #判断是否符合筛选条件
        result.append(x) #大于等于零就将该元素加入结果列表中
print(result) #打印输出

在python 中还有更加简洁高效的方法：

1、filter函数： 

from random import randint
#生成一个包含10个随机数的列表，每个元素的取值范围在-10到10之间
data = [randint(-10, 10) for _ in range(10)] 
#filter函数的第一个参数是过虑函数，通常使用lambda匿名函数，第二个参数是要过滤的列表
result = filter(lambda x: x >=0, data) 
print(result)

 2、列表解析：

...
result = [x for x in data if x >= 0]
...

 

 二、在字典中筛选数据

在字典中根据值来筛选数据，下面的例题是筛选出分数大于等于90分同学的代码：

from random import randint
#同学姓名列表，用字母简化代表同学姓名
students = ['x', 'y', 'z', 'a', 'b', 'c', 'g', 'k', 'j']
#迭代循环姓名列表，随机生成每个同学的分数
data = {s: randint(60, 100) for s in students}
#根据分数来筛选过滤字典中的每个元素
result = {k:v for k, v in data.items() if v >= 90}
print(result)

 三、在集合中筛选数据

在集合中筛选数据的操作和字典非常相似，只是略去键值之间的冒号。下面是在集合中筛选出能整除3的元素的代码：

from random import randint
#生成包含20个随机数的列表，元素的取值范围为1到100之间
data = [randint(1, 100) for _ in range(20)]
#将列表转换为集合，去除重复元素
my_set = set(data)
#筛选出能整除3的元素
result = [x for x in my_set if x%3 == 0]
print(result)