如何将一个列表中的元素插入到另一个列表中，生成所有可能的组合

Question

大家好，我需要一些关于 python 列表的帮助。假设我有两个列表。a = [3,1,5]b = [2,4]我想要的是连续插入列表 b 的元素（不改变 a 的顺序）生成新列表。例如。ans = [[2,4,3,1,5],[2,3,4,1,5],[2,3,1,4,5],[2,3,1,5,4],[3,2,4,1,5]...]感谢您的帮助，我希望我能够正确表达自己。

Answer 1

这不是最干净的方法，但这就是我的意思：

from itertools import permutations

a = [3,1,5]

b = [2,4]

def a_order_is_same(perm):

    i3, i1, i5 = perm.index(3), perm.index(1), perm.index(5)

    return i3 < i1 < i5

ans = [p for p in permutations(a+b) if a_order_is_same(p)]

for p in ans:

    print(p)

--------------------------------------------------

(3, 1, 5, 2, 4)

(3, 1, 5, 4, 2)

(3, 1, 2, 5, 4)

(3, 1, 2, 4, 5)

(3, 1, 4, 5, 2)

(3, 1, 4, 2, 5)

(3, 2, 1, 5, 4)

(3, 2, 1, 4, 5)

(3, 2, 4, 1, 5)

(3, 4, 1, 5, 2)

(3, 4, 1, 2, 5)

(3, 4, 2, 1, 5)

(2, 3, 1, 5, 4)

(2, 3, 1, 4, 5)

(2, 3, 4, 1, 5)

(2, 4, 3, 1, 5)

(4, 3, 1, 5, 2)

(4, 3, 1, 2, 5)

(4, 3, 2, 1, 5)

(4, 2, 3, 1, 5)

Answer 2

我给你另一种选择。

import itertools 

                        

a = [3,1,5]

b = [2,4]

            

c = [b +[a]][0] #0 to get only one level of nested

perm = [x for x in itertools.permutations(c, 3)] 

            

ans = []

这是获得“ans”输出的公式：

for i in range(len(perm)): 

            

    groups = perm[i] #this is the subset like [2, 4, [3, 1, 5]]

    #flatten the list or returns integer

    clean = [y for x in groups for y in (x if isinstance(x,list) else [x])]

    ans.append(clean)

    print(clean)

[[2, 4, 3, 1, 5], [2, 3, 1, 5, 4], [4, 2, 3, 1, 5], [4, 3, 1, 5, 2], [3, 1, 5, 2, 4], [3, 1, 5, 4, 2]]