以多种方式合并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