mirror of
https://github.com/python/cpython.git
synced 2025-08-04 17:08:35 +00:00
Neaten-up the itertools recipes.
This commit is contained in:
parent
fbf66bd617
commit
f5a2e47ad3
2 changed files with 8 additions and 12 deletions
|
@ -563,12 +563,12 @@ which incur interpreter overhead.
|
|||
return zip(a, b)
|
||||
|
||||
def grouper(n, iterable, fillvalue=None):
|
||||
"grouper(3, 'abcdefg', 'x') --> ('a','b','c'), ('d','e','f'), ('g','x','x')"
|
||||
"grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx"
|
||||
args = [iter(iterable)] * n
|
||||
return zip_longest(*args, fillvalue=fillvalue)
|
||||
|
||||
def roundrobin(*iterables):
|
||||
"roundrobin('abc', 'd', 'ef') --> 'a', 'd', 'e', 'b', 'f', 'c'"
|
||||
"roundrobin('ABC', 'D', 'EF') --> A D E B F C"
|
||||
# Recipe credited to George Sakkis
|
||||
pending = len(iterables)
|
||||
nexts = cycle(iter(it).__next__ for it in iterables)
|
||||
|
@ -588,10 +588,8 @@ which incur interpreter overhead.
|
|||
yield set(x for m, x in pairs if m&n)
|
||||
|
||||
def compress(data, selectors):
|
||||
"compress('abcdef', [1,0,1,0,1,1]) --> a c e f"
|
||||
decorated = zip(data, selectors)
|
||||
filtered = filter(operator.itemgetter(1), decorated)
|
||||
return map(operator.itemgetter(0), filtered)
|
||||
"compress('ABCDEF', [1,0,1,0,1,1]) --> A C E F"
|
||||
return (d for d, s in izip(data, selectors) if s)
|
||||
|
||||
def combinations_with_replacement(iterable, r):
|
||||
"combinations_with_replacement('ABC', 3) --> AA AB AC BB BC CC"
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue