python - 以多种方式合并 2 个列表 - Python

Question

我一直在试验多种技术，但我确信有一种顺利的方法可以完成这项工作。

假设我有两个列表，其中包含相同数量的项目(每个 4 个):

a = ['a', 'b', 'c', 'd']    
b = [1, 2, 3, 4]

我想以所有可能的方式合并这些列表，同时保持顺序。示例输出:

a, b, c, d, 1, 2, 3, 4    
1, 2, 3, 4, a, b, c, d    
a, b, 1, 2, c, 3, 4, d

要点是每个列表都必须保留其顺序，因此考虑到它在列表中的位置，一个项目不能在输出中排在另一个项目之前。所以例如输出不能是:

a, b, **d**, c, 1...   > d precedes c whereas c is before d in the original list
1, **4**, a, b, 3....  > 4 precedes 3 whereas 3 is before 4 in the original list

我想这个想法是以所有可能的方式将第二个列表合并到第一个列表中。一个完整的例子是这样的:

a = [a, b]    
b = [1, 2]

期望的输出:

ab12                                                                      
a1b2                                                                             
a12b                                                                         
1ab2                                                                             
1a2b                                                                          
12ab

我该怎么做呢？ itertools 是否有能力以某种方式做到这一点？还是有另一种方法来完成这项工作？请帮忙!

Answer 1

在 2x4 的情况下，您希望在不打乱每个四边形内的顺序的情况下获取所有 8 个元素。这些例子:

a, b, c, d, 1, 2, 3, 4    
1, 2, 3, 4, a, b, c, d    
a, b, 1, 2, c, 3, 4, d

可以转换为“指令”序列，即要从中获取的列表，0 或 1:

0 0 0 0 1 1 1 1
1 1 1 1 0 0 0 0
0 0 1 1 0 1 1 0

一旦你意识到这一点，你可能会注意到我们需要生成的序列都是四个零和四个一的排列。实现这一飞跃后，我们可以使用 itertools:

itertools.permutations([0,0,0,0,1,1,1,1])

对于 2x4 的情况，这给出了 40320 个结果，但只有 70 个不同的结果(因为 itertools.permutations 认为 1,1,1 不同于 1,1,1 如果数字被重新排序) .您可以从此处的答案中获得独特的排列:https://stackoverflow.com/a/6285330/4323或者只使用 set()。

---

将所有这些放在一起，这是一个完整的解决方案:

import itertools

def combos(*seqs):
    counts = map(len, seqs)
    base = []
    for ii, count in enumerate(counts):
        base.extend([ii]*count)
    for take in set(itertools.permutations(base)):
        result = []
        where = [0] * len(seqs)
        for elem in take:
            result.append(seqs[elem][where[elem]])
            where[elem] += 1
        yield result

你可以这样测试(给出 70 个结果):

a = ['a', 'b', 'c', 'd']
b = [1, 2, 3, 4]

for res in combos(a, b):
    print res