Python强化训练笔记(一)——在列表,字典,集合中筛选数据

列表,字典,集合中根据条件筛选数据,如下所示

列表:[-10,2,2,3,-2,7,6,9] 找出所有的非负数

字典:{1:90,2:55,3:87...} 找出所有值大于60的键值对

集合:{2,3,8,6,7,5} 找出所有被3整除的数


列表

>>> from random import randint
>>> data = [randint(-10,10) for x in xrange(10)]
>>> data
[-1, 8, -9, 9, 10, -2, 6, -8, -2, 10]

对于列表来说,想要对其中的数值进行筛选,最先想到的应该是利用遍历

>>> res = []
>>> for x in data:
>>>    if x >= 0:
>>>        res.apend(x)
>>> res
[8,9,10,6,10]

 较为常用的方法还有使用过滤函数filter

>>> filter(lambda x: x >= 0, data)
[8,9,10,6,10]

也可以使用列表生成式

>>> [x for x in data if x >= 0]
[8, 9, 10, 6, 10]

对比这三种方法,开发中通常使用第三种,列表生成式,因为它的运行效率要比过滤函数filter高,而对于第一种则不会去使用。


字典

假如现有20名同学,他们的序号和成绩以对应形式的字典给出,如下所示:

>>> students = {x: randint(30, 100) for x in xrange(1, 21)}
>>> students
{1: 31, 2: 37, 3: 34, 4: 77, 5: 69, 6: 68, 7: 84, 8: 83, 9: 49, 10: 60, 11: 39,
12: 96, 13: 79, 14: 80, 15: 90, 16: 75, 17: 38, 18: 61, 19: 97, 20: 66}

 现在要筛选出成绩优秀(即80分以上)的同学。

使用字典解析

>>> {k:v for k,v in students.iteritems() if v >= 80}
{7: 84, 8: 83, 12: 96, 14: 80, 15: 90, 19: 97}

这个过程中,使用了students.iteritems()而不只是students,这是由于如果只遍历students则只会得到键,而不会得到值。


集合

集合解析:

>>> data
[-1, 8, -9, 9, 10, -2, 6, -8, -2, 10]
>>> s = set(data)
>>> s
set([6, 8, 9, 10, -1, -9, -8, -2])
>>> {x for x in s if x % 3 == 0}
set([9, 6, -9])

 

posted @ 2016-10-09 13:15  大地的谎言  阅读(3349)  评论(0编辑  收藏  举报