cookbook 1.16

大马强

筛选序列中的元素

问题描述：
        我们想对字典中的值做删减或提取

一、推导式

1. # 列表推导式
   
2. datas = [-1, 2, 4, -2, 5, -4, 7, -3, 9, -6, 1, 0]
   
3. print([i for i in datas if i > 0])
   
4. # [2, 4, 5, 7, 9, 1]
   
5. 
   
6. # 生成器推导式
   
7. a = (i for i in datas if i > 0)
   
8. for i in a:
   
9.     print(i, end="")
   
10. print()

复制代码

[2, 4, 5, 7, 9, 1]
245791

二、filter函数
filter(fun, iterable)

• 1、fun=None => [item for in iterable if item]
• 2、fun!=None => [item for in iterable if fun(item)]
  

1. datas = ['1', '2', '3', 'a', 'n/z', '4']
   
2. 
   
3. 
   
4. def is_int(data):
   
5.     try:
   
6.         var = int(data)
   
7.         return True
   
8.     except ValueError:
   
9.         return False
   
10. 
    
11. 
    
12. print(list(filter(is_int, datas)))
    

复制代码

['1', '2', '3', '4']

三、compress函数

1. 
   
2. from itertools import compress
   
3. # compress(iterable, iterable_bool) 返回一个迭代器-
   
4. datas = [-1, 2, 4, -2, 5, -4, 7, -3, 9, -6, 1, 0]
   
5. filter_ = [n > 0 for n in datas]
   
6. print(filter_)
   
7. # [False, True, True, False, True, False, True, False, True, False, True, False]
   
8. print(list(compress(datas, filter_)))

复制代码

[2, 4, 5, 7, 9, 1]

列表推导式与if else的搭配
    1、if
            [i for i in data if i]
    2、if else
            [i if i else 0 for i in datas]