[Python] filter()函数

filter()函数用于过滤序列，过滤掉不符合条件的元素，返回一个迭代器对象

提示: 如果要转换为列表，可以使用list()来转换

语法

filter(function, iterable)

filter()函数接收两个参数，第一个为函数，第二个为序列，序列的每个元素作为参数传递给function函数进行判断，然后返回True或False，最后将返回True的元素放到新列表中 

参数

function: 判断函数

iterable: 可迭代对象

例1: 过滤出列表中的所有奇数 

# 过滤出列表中的所有奇数
def fn(n):
    return n % 2 == 1

tmplist = filter(fn, [1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
list(tmplist) # [1, 3, 5, 7, 9]

例2: 过滤出1~100中平方根是整数的数 

# 过滤出1~100中平方根是整数的数
import math
def is_sqrt(x):
    return math.sqrt(x) % 1 == 0
 
tmplist = filter(is_sqrt, range(1, 101))
list(tmplist) # [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

例3: 过滤序列中的空字符串

# 过滤序列中的空字符串
def not_empty(str):
    return str and str.strip()

# ['A', 'B', 'C']
list(filter(not_empty, ['A', '', 'B', None, 'C', '  ']))