Powersets in Python using itertools

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.

Answers:

Thank you for visiting the Q&A section on Magenaut. Please note that all the answers may not help you solve the issue immediately. So please treat them as advisements. If you found the post helpful (or not), leave a comment & I’ll get back to you as soon as possible.

Method 1

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.

Method 2

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

Method 3

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.

All methods was sourced from stackoverflow.com or stackexchange.com, is licensed under cc by-sa 2.5, cc by-sa 3.0 and cc by-sa 4.0