Python 用所有可能的排列合并两个列表

Question

我正在尝试找出将两个列表合并为所有可能组合的最佳方法。所以，如果我从这样的两个列表开始：

list1 = [1, 2]
list2 = [3, 4]

生成的列表如下所示：

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

也就是说，它基本上生成一个列表列表，以及两者之间的所有可能组合。

我一直在研究 itertools，我很确定它能找到答案，但我想不出一种方法让它以这种方式运行。我最接近的是：

list1 = [1, 2, 3, 4]
list2 = [5, 6, 7, 8]
print list(itertools.product(list1, list2))

产生了：

[(1, 5), (1, 6), (1, 7), (1, 8), (2, 5), (2, 6), (2, 7), (2, 8), (3, 5), (3, 6), (3, 7), (3, 8), (4, 5), (4, 6), (4, 7), (4, 8)]

所以它会在每个列表中执行所有可能的项目组合，但不会执行所有可能的结果列表。我怎样才能做到这一点？

编辑：最终目标是能够单独处理每个列表以确定效率（我正在处理的实际数据更复杂）。因此，在上面的原始示例中，它会像这样工作：

list1 = [1, 2]
list2 = [3, 4]

Get first merged list: [[1,3], [2, 4]]
    Do stuff with this list
Get second merged list: [[1,4], [2, 3]]
    Do stuff with this list

如果我得到上面描述的 "list of lists of lists" 输出，那么我可以将它放入 for 循环并继续处理。其他形式的输出也可以，但它似乎是最简单的。

Answer 1

编辑了我的代码以提供您想要的输出。

list1 = [1,2]
list2 = [3,4]
combined = []

for a in list1:
    new_list = []
    for b in list2:
        new_list.append([a, b])
    combined.append(new_list)

print combined

Answer 2

您可以通过构造两个列表成员的所有排列来创建列表，包含列表组合。

lst1 = [1,2]
lst2 = [3,4]

#lst = [[j,k] for j in lst1 for k in lst2] # [[1,3],[1,4],[2,3],[2,4]]
lst = [[[j,k] for j in lst1] for k in lst2] # [[[1,3],[2,3]],[[1,4],[2,4]]]
print lst

Answer 3

试试这个：

combos=[]
for i in list1:
      for j in list2:
          combos.append([i,j])
print combos

Answer 4

repeat 第一个列表，permutate 第二个和 zip 全部一起

>>> from itertools import permutations, repeat
>>> a = [1, 2, 3]
>>> b = [4, 5, 6]
>>> list(list(zip(r, p)) for (r, p) in zip(repeat(a), permutations(b)))
[[(1, 4), (2, 5), (3, 6)],
 [(1, 4), (2, 6), (3, 5)],
 [(1, 5), (2, 4), (3, 6)],
 [(1, 5), (2, 6), (3, 4)],
 [(1, 6), (2, 4), (3, 5)],
 [(1, 6), (2, 5), (3, 4)]]

编辑：正如 Peter Otten 指出的那样，内部 zip 和 repeat 是多余的。

[list(zip(a, p)) for p in permutations(b)]

Answer 5

尝试使用列表生成器创建嵌套列表：

>>> [[[x,y] for x in list1] for y in list2]
[[[1, 3], [2, 3]], [[1, 4], [2, 4]]]
>>>

或者，如果您想要单行列表，只需删除括号：

>>> [[x,y] for x in list1 for y in list2]
[[1, 3], [1, 4], [2, 3], [2, 4]]

Answer 6

接受的答案可以简化为

a = [1, 2, 3]
b = [4, 5, 6]
[list(zip(a, p)) for p in permutations(b)]

(Python 2中可以省略list()调用)

Answer 7

由于@pacholik 的回答没有涵盖不同长度的列表，这是我的解决方案，使用带有两个变量的列表理解：

first_list = [1, 2, 3]
second_list = ['a', 'b']

combinations = [(a,b) for a in first_list for b in second_list]

输出如下所示：

[(1, 'a'), (1, 'b'), (2, 'a'), (2, 'b'), (3, 'a'), (3, 'b')]

Python 用所有可能的排列合并两个列表

Python merging two lists with all possible permutations

python

list

itertools