Powersets in Python using itertools

Question

I'm trying to create a powerset in Python 3. I found a reference to the itertools module, and I've used the powerset code provided on that page. The problem: the code returns a reference to an itertools.chain object, whereas I want access to the elements in the powerset. My question: how to accomplish this?

Many thanks in advance for your insights.

related: [what's a good way to combinate through a set?](http://stackoverflow.com/q/1482308/4279) — jfs, Oct 30 '16 at 22:52

Martijn Pieters · Accepted Answer · 2017-11-18T22:17:48.410

itertools functions return iterators, objects that produce results lazily, on demand.

You could either loop over the object with a for loop, or turn the result into a list by calling list() on it:

from itertools import chain, combinations

def powerset(iterable):
    "powerset([1,2,3]) --> () (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)"
    s = list(iterable)
    return chain.from_iterable(combinations(s, r) for r in range(len(s)+1))

for result in powerset([1, 2, 3]):
    print(result)

results = list(powerset([1, 2, 3]))
print(results)

You can also store the object in a variable and use the next() function to get results from the iterator one by one.

score 3 · Answer 2 · answered Dec 11 '18 at 15:01

3

Here's a solution using a generator:

from itertools import combinations

def all_combos(s):
    n = len(s)
    for r in range(1, n+1):
        for combo in combinations(s, r):
            yield combo

answered Dec 11 '18 at 15:01

Allen Downey

134
5

score 2 · Answer 3 · answered Jul 29 '20 at 01:47

Allen Downey's answer is what I would use, except I would replace the range with range(n+1) since the empty set is considered an element of the power set.

This code can easily be rewritten as a one liner:

(combo for r in range(len(s) + 1) for combo in combinations(s, r))

Your choice whether to surround it with () for an iterator or [] for a list.

Powersets in Python using itertools

3 Answers3

Linked

Related