How to apply itertools.product to elements of a list of lists? [duplicate]

Question

I have a list of arrays and I would like to get the cartesian product of the elements in the arrays.

I will use an example to make this more concrete...

itertools.product seems to do the trick but I am stuck in a little detail.

arrays = [(-1,+1), (-2,+2), (-3,+3)];

If I do

cp = list(itertools.product(arrays));

I get

cp = cp0 = [((-1, 1),), ((-2, 2),), ((-3, 3),)]

But what I want to get is

cp1 = [(-1,-2,-3), (-1,-2,+3), (-1,+2,-3), (-1,+2,+3), ..., (+1,+2,-3), (+1,+2,+3)].

I have tried a few different things:

cp = list(itertools.product(itertools.islice(arrays, len(arrays)))); cp = list(itertools.product(iter(arrays, len(arrays))));

They all gave me cp0 instead of cp1.

Any ideas?

Thanks in advance.

You already have the answers below, but this has some good use cases for the itertools product function: hackerrank.com/challenges/itertools-product — Afflatus
– Afflatus, Commented Sep 16, 2016 at 19:31

interjay · Accepted Answer · 2010-06-13 21:44:57Z

>>> list(itertools.product(*arrays)) [(-1, -2, -3), (-1, -2, 3), (-1, 2, -3), (-1, 2, 3), (1, -2, -3), (1, -2, 3), (1, 2, -3), (1, 2, 3)]

This will feed all the pairs as separate arguments to product, which will then give you the cartesian product of them.

The reason your version isn't working is that you are giving product only one argument. Asking for a cartesian product of one list is a trivial case, and returns a list containing only one element (the list given as argument).

rkhayrov · Accepted Answer · 2010-06-13 21:41:14Z

40

>>> arrays = [(-1,+1), (-2,+2), (-3,+3)] >>> list(itertools.product(*arrays)) [(-1, -2, -3), (-1, -2, 3), (-1, 2, -3), (-1, 2, 3), (1, -2, -3), (1, -2, 3), (1, 2, -3), (1, 2, 3)]

answered Jun 13, 2010 at 21:41

rkhayrov

10.3k2 gold badges38 silver badges42 bronze badges

2 Comments

Ominus Over a year ago

does having an asterix prefix an array have special meaning in python or is this just something specific to itertools.product?

Franck Dernoncourt Over a year ago

See stackoverflow.com/q/5239856/395857 and Unpacking Argument Lists.

Wojciech Moszczyński · Accepted Answer · 2020-05-15 15:21:09Z

you can do it in three rurch using itertools.product

lst=[] arrays = [(-1,+1), (-2,+2), (-3,+3)] import itertools for i in itertools.product(*arrays): lst.append(i) print(lst)

Collectives™ on Stack Overflow

How to apply itertools.product to elements of a list of lists? [duplicate]

3 Answers 3

Comments

2 Comments

Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

2 Comments

Comments

Linked

Related