Question

例如，

list = [0, 1, 2]

我想要一份所有可能的2种组合的列表：

combinations = [(0,0), (0,1), (0,2), (1,0), (1,1), (1,2), (2,0), (2,1), (2,2)]

在我看来，Python中的itertools中的所有工具只能生成（1,0）和（0,1）中的一个，而不是两者，我需要两者。除了手工输入之外的任何建议吗？

Answer 1

您正在寻找该列表中的笛卡尔积，而不是排列或组合。因此，您应将itertools.product与repeat=2：

一起使用

from itertools import product

li = [0, 1, 2]
print(list(product(li, repeat=2)))
>> [(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]

Answer 2

可以通过导入itertools来完成：

import itertools

list1 = [0, 1, 2]
print(list(itertools.product(list1,repeat=2)))

输出：

[(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]

资源：你可以了解更多 - here

如何从列表中的元素生成所有可能的组合？

2 个答案: