All conditional permutations of a nested list in python

Question

I have an array like bellow (A) and I want to convert it to (B).

A:

A = [['a', 'ae', 'oa'], ['a'], ['l'], ['y', 'i']]

B:

B = ['aaly',
'aali',
'aealy',
'aeali',
'oaaly',
'oaali']

This might help you https://stackoverflow.com/questions/798854/all-combinations-of-a-list-of-lists — Rashid 'Lee' Ibrahim, Oct 28 '21 at 20:05

Dani Mesejo · Accepted Answer · 2021-10-28T20:14:28.550

2

Use:

from itertools import product

A = [['a', 'ae', 'oa'], ['a'], ['l'], ['y', 'i']]

res = ["".join(p) for p in product(*A)]
print(res)

Output

['aaly', 'aali', 'aealy', 'aeali', 'oaaly', 'oaali']

As an alternative use:

res = list(map("".join, product(*A)))

edited Oct 28 '21 at 20:14

answered Oct 28 '21 at 20:06

Dani Mesejo

1 Answers1