的Python模擬對Ruby可枚舉.partition方法

Question

可能重複：
python equivalent of filter() getting two output lists (i.e. partition of a list)

是否有任何內置函數或Python標準庫，也許一些模塊，從模擬Enumerable.partition行爲Ruby並遍歷一個對象只需要一次，以獲得兩個基於謂詞函數傳遞的列表/元組？

Answer 1

被盜無恥地從this question - 您可以使用tee功能從itertools：

from itertools import tee 

def split_on_condition(seq, condition): 
    l1,l2 = tee((condition(item),item) for item in seq) 
    return (i for p, i in l1 if p), (i for p, i in l2 if not p)