Python: Combination in lists of lists (?)

Question

First of all I wanted to say that my title is probably not describing my question correctly. I don't know how the process I am trying to accomplish is called, which made searching for a solution on stackoverflow or google very difficult. A hint regarding this could already help me a lot!

What I currently have are basically two lists with lists as elements. Example:

List1 = [ [a,b], [c,d,e], [f] ]
List2 = [ [g,h,i], [j], [k,l] ]

These lists are basically vertices of a graph I am trying to create later in my project, where the edges are supposed to be from List1 to List2 by rows.

If we look at the first row of each of the lists, I therefore have:

[a,b] -> [g,h,i]

However, I want to have assingments/edges of unique elements, so I need:

[a] -> [g]
[a] -> [h]
[a] -> [i]
[b] -> [g]
[b] -> [h]
[b] -> [i]

The result I want to have is another list, with these unique assigments as elements, ie

List3 = [ [a,g], [a,h], [a,i], [b,g], ...]

Is there any elegant way to get from List1 and List2 to List 3?

The way I wanted to accomplish that is by going row by row, determining the amount of elements of each row and then write clauses and loops to create a new list with all combinations possible. This, however, feels like a very inefficient way to do it.

Answer 1

You can zip your two lists, then use itertools.product to create each of your combinations. You can use itertools.chain.from_iterable to flatten the resulting list.

>>> import itertools
>>> List1 = [ ['a','b'], ['c','d','e'], ['f'] ]
>>> List2 = [ ['g','h','i'], ['j'], ['k','l'] ]
>>> list(itertools.chain.from_iterable(itertools.product(a,b) for a,b in zip(List1, List2)))
[('a', 'g'), ('a', 'h'), ('a', 'i'), ('b', 'g'), ('b', 'h'), ('b', 'i'), ('c', 'j'), ('d', 'j'), ('e', 'j'), ('f', 'k'), ('f', 'l')]

Answer 2

If you don't want to use itertools, you can also use list comprehensions in combination with zip to do this fairly elegantly:

lst1 = [['a','b'], ['c','d','e'], ['f']]
lst2 = [['g','h','i'], ['j'], ['k','l']]
edges = [[x, y] for il1, il2 in zip(lst1, lst2) for x in il1 for y in il2]

Answer 3

import itertools
k = []
for a,b in zip(List1,List2):
    for j in itertools.product(a,b):
        k.append(j)
print k