如何爲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解決方案仍然具有相當的可讀性，但在許多情況下，嘗試經歷許多麻煩來定義正確的功能並不像僅僅執行操作一樣有價值。