Python filter的用法

filter 过滤器

参考官方解释： https://docs.python.org/3.7/library/functions.html#filter
filter(function, iterable): 从函数中返回true的元素并生成迭代器
如果函数不为None，则返回True的元素。
如果函数为None，则返回所有本身可以判断为True的元素。

1. 函数不为None, 则返回true的元素,即函数中返回结果为true的x的值

print(list(filter(lambda x: x % 3 == 0, range(10))))  # --> [0, 3, 6, 9]
print(list(filter(lambda x: x if x % 3 == 0 else None, range(10))))  # --> [3, 6, 9]

分析第一个语句：lambda x: x % 3 == 0, 其相当于以下语句，返回的结果是[True, False, False, True, False, False, True, False, False, True]. 所以整个第一个语句filter只返回true的元素，返回的结果是[0,3,6,9]

def is_third_div(x):
    return x % 3 == 0


list1 = []
for i in range(10):
    list1.append(is_third_div(i))
print(list1)  # [True, False, False, True, False, False, True, False, False, True]

分析第2个语句：lambda x: x if x % 3 == 0 else None，相当于以下语句,lamda 函数返回结果[0, None, None, 3, None, None, 6, None, None, 9]
因为python认为0，None是False,可参考list3返回的列表[3,6,9],所以第二个语句返回的结果是[3,6,9]

def add_fun(x):
    if x % 3 == 0:
        return x
    else:
        return None


list2 = []
for i in range(10):
    list2.append(add_fun(i))

print(list2)    # [0, None, None, 3, None, None, 6, None, None, 9]
list3 = []
for i in list2:
    if i:
        list3.append(i)
print(list3)  # [3, 6, 9]

2. 如果函数为None，则返回所有本身可以判断为True的元素。可以看出负数没过滤掉，0过滤掉了，

print(list(filter(None, range(-1, 5))))  # [-1, 1, 2, 3, 4]