如何遍历数据框以打印两个列表列的所有可能组合？

Question

Lets say I have a datavframe:

Column1	Column2
1	A
2	A, B
3	A, B, C

I want to print every possible combination from the two lists on a new row. The output should look something like:

1A
2A 
2B
3A 
3B 
3C

Answer 1

Let's try split Column2 into list then explode Column2 . At last join two columns and convert to list.

out = (df.assign(Column2=df['Column2'].str.split(', '))
       .explode('Column2')
       # [['Column1', 'Column2']] # uncomment this line if there are more than two target columns
       .astype(str)
       .agg(''.join, axis=1)
       .tolist())

print(out)

['1A', '2A', '2B', '3A', '3B', '3C']

Answer 2

Better use pure python here:

from itertools import product
out = [''.join(x) for a,b in zip(df['Column1'], df['Column2'])
       for x in product([str(a)], b.split(', '))]

output: ['1A', '2A', '2B', '3A', '3B', '3C']

Classical loop:

for a,b in zip(df['Column1'], df['Column2']):
    for x in product([str(a)], b.split(', ')):
        print(''.join(x))

output:

1A
2A
2B
3A
3B
3C

Answer 3

Another possible solution:

df.apply(lambda x: [str(x.Column1) + item for item in x.Column2.split(', ')], axis=1).explode()

Answer 4

Another way to use itertools.product :

from itertools import product
import re
out = df.apply(lambda x: list(
                product([str(x['Column1'])], list(re.sub('\s*,\s*', '', x['Column2'])))),
               axis=1)
out = [''.join(b) for a in out for b in a]

print(out):

['1A', '2A', '2B', '3A', '3B', '3C']