使用itertools从2个列表生成组合

Question

我有两个清单[1,2,3]和[4,5,6].我想使用以下生成所有组合的列表itertools

ResultingList = [[1,4],[1,5],[1,6],[2,4],[2,5],[2,6],[3,4],[3,5],[3,6]]

到目前为止,我只调查了itertools.combinations函数,它似乎只能处理这样的事情:

list(itertools.combinations([1,2,3,4,5,6],2))

哪个输出不正确的结果.我怎样才能生成ResultingList以上内容？

谢谢

Answer 1

使用产品:

>>> from itertools import product
>>> list(product([1,2,3], [4,5,6]))
[(1, 4), (1, 5), (1, 6), (2, 4), (2, 5), (2, 6), (3, 4), (3, 5), (3, 6)]

一般理解:

如文档product中所述,相当于: ((x,y) for x in A for y in B)其中A和B是您的输入列表