如何为Python itertools.ifilter函数提供额外的参数？

Question

在Python中，我有以下功能：

def is_a_nice_element(element, parameter): 
    #do something 
    return True or False 

现在我想过滤列表，使用此功能作谓语，给出一个固定的参数。 Python有itertools.ifilter函数，但我不知道如何传递参数。这可能吗？如果不是，我该如何解决这个问题？

Answer 1

把它包在一个lambda：

itertools.ifilter(lambda e: is_a_nice_element(e, 42), iterable) 

42是你的额外的参数，或者其他任何你希望它是。

Answer 2

如果您使用的是ifilter，则参数需要保持不变，在这种情况下，您可以使用默认参数parameter=something。如果你想让参数变化，你需要使用另一种方法来进行二元谓词。

如果您已经拥有清单，ifilter相对于内置的filter有点矫枉过正。

描述你想要完成什么将有助于做出更具体的答案。

Answer 3

我喜欢functools.partial比lambda好得多。

itertools.ifilter(partial(is_a_nice_element, parameter=X), iterable) 

Answer 4

该解决方案使用lambda和functools.partial是十分正确的，直接回答你的问题，但我不认为他们是真正理想的解决方案。

使用filter/itertools.ifilter/map/itertools.imap与匿名功能prettymuch总是会比使用生成的表达或列表理解不太清楚。例如，在这种情况下，我会写发电机表达

(item for item in iterable if is_a_nice_element(item, constant)) 

代替使用ifilter。在这种情况下，ifilter解决方案仍然具有相当的可读性，但在许多情况下，尝试经历许多麻烦来定义正确的功能并不像仅仅执行操作一样有价值。