Python Iterator Partitioner

partition_iterator.py

import itertools

def partition(items, predicate=int, n=2):
    """ Partition an iterable of `items` into `n` streams by `predicate`
    """
    def filterer(i, tee):
        return (item for pred, item in tee if pred == i)
    pred_items = ((predicate(item), item) for item in items)
    teed_items = enumerate(itertools.tee(pred_items, n))
    return (filterer(x, t) for x, t in teed_items)