Skip to content

Instantly share code, notes, and snippets.

@Richard-Mathie

Richard-Mathie/partition_iterator.py

Created November 8, 2018 17:33
Show Gist options
  • Save Richard-Mathie/419779ce26307e43171c71b515afb6b5 to your computer and use it in GitHub Desktop.
Save Richard-Mathie/419779ce26307e43171c71b515afb6b5 to your computer and use it in GitHub Desktop.
Download ZIP
Python Iterator Partitioner
Raw
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)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment