生成 python 中 3 个一组的 2 个列表元素的组合

Question

我在下面有 2 个列表

list_1 = ['A','B','C']
list_2 = ['D','E','F']

尝试生成组合以复制以下 output

(AD, BE, CF), (AD, BF, CE), (AE, BD, CF), (AE, BF, CD), (AF, BD, CE) (AF, BE, CD)

到目前为止，我已经能够使用 itertools 创建组合。

list = [list_1, list_2]

combination = [p for p in itertools.product(*list)]

这给了我

[('A', 'D'),
 ('A', 'E'),
 ('A', 'F'),
 ('B', 'D'),
 ('B', 'E'),
 ('B', 'F'),
 ('C', 'D'),
 ('C', 'E'),
 ('C', 'F')]

但不确定如何进一步进行。

Answer 1

只有zip第一个列表和第二个列表的所有排列：

from itertools import permutations

def combs(l1, l2):
    for p in permutations(l2, len(l2)):
        yield list(map(''.join, zip(l1, p)))
        # or simpler:
        # yield [a+b for a, b in zip(l1, p)]

>>> list(combs(list_1, list_2))
[['AD', 'BE', 'CF'], ['AD', 'BF', 'CE'], ['AE', 'BD', 'CF'], ['AE', 'BF', 'CD'], ['AF', 'BD', 'CE'], ['AF', 'BE', 'CD']]

一些文档：

• zip
• map
• str.join
• itertools.permutations
• 生成器函数（ yield关键字）