Question

I have the following data:

data = [['AB', 'BS, BT'], ['AH', 'AH'], ['AS', 'AS, GS']]

I would like to iterate through the lists of list to produce a list of tuples.

new_data = [('AB', 'BS'), ('AB', 'BT'), ('AH', 'AH'), ('AS', 'AS') ('AS', 'GS')]

I was thinking about using the zip() function, but wasn’t sure if I was using the right logic.

Answer 1

zip wouldn’t be my first choice. My first choice would be itertools.product and itertools.chain

In [103]: data = [['AB', 'BS, BT'], ['AH', 'AH'], ['AS', 'AS, GS']]

In [104]: [list(itertools.product([d[0]], d[1].split(','))) for d in data]
Out[104]: [[('AB', 'BS'), ('AB', ' BT')], [('AH', 'AH')], [('AS', 'AS'), ('AS', ' GS')]]

In [105]: list(itertools.chain.from_iterable([list(itertools.product([d[0]], d[1].split(','))) for d in data]))
Out[105]: [('AB', 'BS'), ('AB', ' BT'), ('AH', 'AH'), ('AS', 'AS'), ('AS', ' GS')]

Hope this helps