Given the following lists:
a = ['a','b']
b = [1,2,3,4]
I'd like to produce this:
c = ['a1','a2','a3','a4','b1','b2','b3','b4']
So I basically want to join every element of b to each element in a.
I'd like an approach similar to this:
[x+str(y) for x in a and y in b]
Thanks in advance!
a = ['a','b']
b = [1,2,3,4]
c = [x+str(y) for x in a for y in b]
print(c)
You need to iterate twice within the list comprehension as:
>>> a = ['a','b']
>>> b = [1,2,3,4]
>>> [i+str(j) for i in a for j in b]
['a1', 'a2', 'a3', 'a4', 'b1', 'b2', 'b3', 'b4']
You can also mix list comprehension with generation of combinations. Itertools module is a good way how to work with combinations.
import itertools
c = [x + str(y) for x, y in itertools.product(a, b)]
Try this, replacing the and in your example with a for to add an extra loop:
[x + str(y) for x in a for y in b]
This will loop through both lists in a single list comprehension.
[str(a1) + str(b1) for a1 in a for b1 in b]
I arrived too late to this question, but here are my two cents for doing this on python 3.6+
[f'{a_item}{b_item}' for a_item in a for b_item in b]
c = [x+str(y) for x in a and y in b]is the solution which you have included in your question, what are you actually asking for?