Ruby Enumerable .partition 方法的 Python 类似物

Question

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

python 标准库中是否有任何内置函数或某些模块，可以模拟 Ruby 中的 Enumerable.partition 行为，并仅对对象进行一次迭代以根据传递的谓词函数获取两个列表/元组？

Answer 1

无耻地从this question窃取-你可以使用itertools的tee函数：

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)