cookbook 1.16

大马强

筛选序列中的元素

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

一、推导式

# 列表推导式
datas = [-1, 2, 4, -2, 5, -4, 7, -3, 9, -6, 1, 0]
print([i for i in datas if i > 0])
# [2, 4, 5, 7, 9, 1]

# 生成器推导式
a = (i for i in datas if i > 0)
for i in a:
    print(i, end="")
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)]
  

datas = ['1', '2', '3', 'a', 'n/z', '4']


def is_int(data):
    try:
        var = int(data)
        return True
    except ValueError:
        return False


print(list(filter(is_int, datas)))

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

三、compress函数

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