python filter内建函数

时间:2021-12-10 19:27:16

以下是filter函数的官方文档,注意最后一段,当function不为None时,函数相似于[item for item in iterable if function(item)],function为None时,函数相似于[item for item in iterable if item]。

filter(function, iterable)
Construct a list from those elements of iterable for which function returns true. iterable may be either a sequence, a container which supports iteration, or an iterator. If iterable is a string or a tuple, the result also has that type; otherwise it is always a list. If function is None, the identity function is assumed, that is, all elements of iterable that are false are removed.
Note that filter(function, iterable) is equivalent to [item for item in iterable if function(item)] if function is not None and [item for item in iterable if item] if function is None.

根据官方文档,为了方便理解,自己试着去实现了下,没有带异常处理。分析filter(function, iterable),满足以下条件

  • function, 为None, 或者函数。
  • iterable, 为list, tuple, 或者basestring。

根据function生成列表,再根据iterable对列表进行相应处理。

def myFilter(func, seq):
    if func is None:
        results = [i for i in seq if i]
    else:
        results = [i for i in seq if func(i)]
    if isinstance(seq, tuple):
        results = tuple(results)
    elif isinstance(seq, basestring):
        results = ''.join(results)
    return results

运行结果与filter自建函数输出一致。

print '-------------------filter------------------------------'
print filter(lambda x: x > 4, range(10))
print filter(lambda x: x > 4, tuple(range(10)))
print filter(None, range(10))
print filter(lambda x: x > '3', '12345')
print filter(None, '01234')
print '-------------------myFilter----------------------------'
print myFilter(lambda x: x > 4, range(10))
print myFilter(lambda x: x > 4, tuple(range(10)))
print myFilter(None, range(10))
print myFilter(lambda x: x > '3', '12345')
print myFilter(None, '01234')