python - Python：生成一个包含多个列表的列表

Question

我有以下列表：

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

从上面，我想生成一个列表，其中包含：

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

这个过程叫什么？

生成python列表的阶乘？

有没有内置库可以做到这一点？

Answer 1

使用itertools.product：

>>> import itertools
>>> [list(xs) for xs in itertools.product([1,2,3], [1], [1,2,3,4])]
[[1, 1, 1], [1, 1, 2], [1, 1, 3], [1, 1, 4], [2, 1, 1], [2, 1, 2], [2, 1, 3], [2, 1, 4], [3, 1, 1], [3, 1, 2], [3, 1, 3], [3, 1, 4]]

Answer 2

itertools.product

>>> lists = [[1,2,3], [1], [1,2,3,4]]  
>>> from itertools import product
>>> map(list, product(*lists))
[[1, 1, 1], [1, 1, 2], [1, 1, 3], [1, 1, 4], [2, 1, 1], [2, 1, 2], [2, 1, 3], [2, 1, 4], [3, 1, 1], [3, 1, 2], [3, 1, 3], [3, 1, 4]]

注意：使用map允许我将product' 迭代的其他元组结果轻松转换为列表。

Answer 3

inputList = [[1,2,3], [1], [1,2,3,4]]

import itertools
print [list(item) for item in itertools.product(*inputList)]

输出

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

Answer 4

正如其他答案itertools.product中所建议的那样，这里是走的路，但为了完整起见并作为说明，itertools.product这里是使用列表理解的解决方案：

result = [[x,y,z] for x in [1,2,3] for y in [1] for z in [1,2,3,4]]

这是使用普通 for 循环的相同内容，这可能使其更具可读性：

result = []
for x in [1,2,3]:
    for y in [1]:
        for z in [1,2,3,4]:
            result.append([x, y, z])