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', ' ']))