python - 列表列表：查找元素的组合

Question

假设我有一个包含 n 个列表的列表。有没有一种巧妙的方法来获得长度为 n 的列表的所有可能组合，每个元素保持它的位置。

这是一个例子：

lists=[[1],[2,3],[4,5]]

输出：

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

Answer 1

您可以itertools.product为此使用：

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

Answer 2

这是首先想到的：

def find_permutations(lists, context):
    if len(lists) == 1:
        for item in lists[0]:
            yield context + [item]
    else:
        for item in lists[0]:
           for permutation in find_permutations(lists[1:], context + [item]):
               yield permutation