Get permutations of lists using python

Question

Let's assume I have list a:

a = [0, 0, [0, 1], [0, 1, 2]]

where a can be of any length, and its constituents can be of any length and I want to generate a new list that can be any combination of the values such that for the given example I would return:

b = [[0,0,0,0],
     [0,0,0,1],
     [0,0,0,2], 
     [0,0,1,0],
     [0,0,1,1],
     [0,0,1,2]]

I think it's just a matter of looping, but I would appreciate any help.

Thanks

falsetru · Accepted Answer · 2014-12-30 16:57:17Z

4

Using itertools.product:

>>> import itertools
>>> a = [0, 0, [0, 1], [0, 1, 2]]
>>> a2 = [x if isinstance(x, list) else [x] for x in a]
>>> #  = [[0], [0], [0, 1], [0, 1, 2]]
>>> list(itertools.product(*a2))
[(0, 0, 0, 0), (0, 0, 0, 1), (0, 0, 0, 2), (0, 0, 1, 0), (0, 0, 1, 1), (0, 0, 1, 2)]

answered Dec 30, 2014 at 16:57

falsetru

371k69 gold badges769 silver badges659 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

MightyPork Over a year ago

is the # comment intentional?

falsetru Over a year ago

@MightyPork, Yes, it is. I want to tell that elements should be sequences (not a int).

Collectives™ on Stack Overflow

Get permutations of lists using python

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related