python - 通过python过滤函数传递参数

Question

我正在尝试按某个关键字过滤单词/短语列表。我发现的所有关于 filter() 函数的例子都使用了数字，所以我想知道这是否可能。我知道如果它调用的函数返回 True，filter() 会将一个项目放入结果列表。

假设我有这样的东西:

def filtCheck(item, filt):
    if filt in item:
        return True

def funct():
    filt = 'Hi'
    set1 = ['Hello, world', 'Hi there', 'Hi friend']
    set2 = filter(filtCheck(filt), set1)

    print set2

这是我感到困惑的地方。我将如何在 set2 行上写第一个参数？显然不是它的编写方式，因为我的 filtCheck 函数有两个参数，而我只提供一个。我是否也需要修改我的 filtCheck 功能？但是如果我从中取出 item 参数，就没有字符串来检查 filt 是否在里面。

Answer 1

你可以使用 lambda (另见 http://docs.python.org/2/reference/expressions.html#lambda ):

set2 = filter(lambda item: filtCheck(item, 'Hi'), set1)

或者您可以使用 functools.partial :

from functools import partial
set2 = filter(partial(filtCheck, filt="Hi"), set1)

或者您可以跳过过滤功能并使用 list comprehension :

set2 = [item for item in set1 if filtCheck(item, "Hi")]

# or simply

set2 = [item for item in set1 if "Hi" in item]