Commit ace67339 authored by Raymond Hettinger's avatar Raymond Hettinger

Backport r68942: update powerset() recipe.

parent 89e12963
...@@ -590,11 +590,9 @@ which incur interpreter overhead. ...@@ -590,11 +590,9 @@ which incur interpreter overhead.
nexts = cycle(islice(nexts, pending)) nexts = cycle(islice(nexts, pending))
def powerset(iterable): def powerset(iterable):
"powerset('ab') --> set([]), set(['a']), set(['b']), set(['a', 'b'])" "powerset([1,2,3]) --> () (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)"
# Recipe credited to Eric Raymond s = list(iterable)
pairs = [(2**i, x) for i, x in enumerate(iterable)] return chain.from_iterable(combinations(s, r) for r in range(len(s)+1))
for n in xrange(2**len(pairs)):
yield set(x for m, x in pairs if m&n)
def compress(data, selectors): def compress(data, selectors):
"compress('ABCDEF', [1,0,1,0,1,1]) --> A C E F" "compress('ABCDEF', [1,0,1,0,1,1]) --> A C E F"
......
...@@ -1277,11 +1277,9 @@ Samuele ...@@ -1277,11 +1277,9 @@ Samuele
... nexts = cycle(islice(nexts, pending)) ... nexts = cycle(islice(nexts, pending))
>>> def powerset(iterable): >>> def powerset(iterable):
... "powerset('ab') --> set([]), set(['a']), set(['b']), set(['a', 'b'])" ... "powerset([1,2,3]) --> () (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)"
... # Recipe credited to Eric Raymond ... s = list(iterable)
... pairs = [(2**i, x) for i, x in enumerate(iterable)] ... return chain.from_iterable(combinations(s, r) for r in range(len(s)+1))
... for n in range(2**len(pairs)):
... yield set(x for m, x in pairs if m&n)
>>> def compress(data, selectors): >>> def compress(data, selectors):
... "compress('ABCDEF', [1,0,1,0,1,1]) --> A C E F" ... "compress('ABCDEF', [1,0,1,0,1,1]) --> A C E F"
...@@ -1379,8 +1377,8 @@ perform as purported. ...@@ -1379,8 +1377,8 @@ perform as purported.
>>> list(roundrobin('abc', 'd', 'ef')) >>> list(roundrobin('abc', 'd', 'ef'))
['a', 'd', 'e', 'b', 'f', 'c'] ['a', 'd', 'e', 'b', 'f', 'c']
>>> list(map(sorted, powerset('ab'))) >>> list(powerset([1,2,3]))
[[], ['a'], ['b'], ['a', 'b']] [(), (1,), (2,), (3,), (1, 2), (1, 3), (2, 3), (1, 2, 3)]
>>> list(compress('abcdef', [1,0,1,0,1,1])) >>> list(compress('abcdef', [1,0,1,0,1,1]))
['a', 'c', 'e', 'f'] ['a', 'c', 'e', 'f']
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment