python3: Can I get cartesian product of 'one' 3d list?

Question

I have a list looks like below

[['a'], ['b'], [[1], [2], [3]], ['d']]

and I need to get

[['a'], ['b'], [1], ['d']], [['a'], ['b'], [2], ['d']], [['a'], ['b'], [3], ['d']]

I saw few posts and found itertools.product, but I think this needs multiple lists in same dimension as argument.

Is there any way to get the result?

`itertools.product(*your_list)` – Aran-Fey May 21 '18 at 06:43 — Aran-Fey, May 21 '18 at 06:43

score 0 · Accepted Answer · answered May 21 '18 at 06:43

You can use itertools.product with the star operator (see here).

That would look like

from itertools import product
A = [['a'], ['b'], [[1], [2], [3]], ['d']]

for a in product(*A):
   print(a)

# yields
# ('a', 'b', [1], 'd')
# ('a', 'b', [2], 'd')
# ('a', 'b', [3], 'd')

python3: Can I get cartesian product of 'one' 3d list?

1 Answers1