Python 中所有 N 个列表的排列组合


如果我们有两个列表,需要将第一个列表的每个元素与第二个列表的每个元素组合,那么我们可以使用以下方法。

使用 For 循环

在这种直接的方法中,我们创建一个包含每个列表元素排列组合的列表列表。我们设计了一个嵌套的 for 循环。内层循环对应第二个列表,外层循环对应第一个列表。

示例

 在线演示

A = [5,8]
B = [10,15,20]

print ("The given lists : ", A, B)
permutations = [[m, n] for m in A for n in B ]

输出

运行以上代码,得到以下结果

The given lists : [5, 8] [10, 15, 20]
permutations of the given values are : [[5, 10], [5, 15], [5, 20], [8, 10], [8, 15], [8, 20]]

使用 itertools

itertools 模块有一个名为 product 的迭代器。它与上述嵌套 for 循环的功能相同。在内部创建嵌套 for 循环以生成所需的乘积。

示例

 在线演示

import itertools

A = [5,8]
B = [10,15,20]

print ("The given lists : ", A, B)
result = list(itertools.product(A,B))
print ("permutations of the given lists are : " + str(result))

输出

运行以上代码,得到以下结果

The given lists : [5, 8] [10, 15, 20]
permutations of the given values are : [(5, 10), (5, 15), (5, 20), (8, 10), (8, 15), (8, 20)]

更新于: 2020年1月7日

859 次浏览

开启你的 职业生涯

通过完成课程获得认证

开始学习
广告