python - 生成嵌套列表的所有可能组合

Question

假设我在 Python 中有以下嵌套列表：

lst = [[0, 1, 2],
       [3],
       [4, 5]]

我想创建一个所有可能的长度组合的lst列表，其中每个列表lst对每个组合只采样一次。所以这个例子的输出是：

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

我可以：

[[x0,x1,x2] for x0 in lst[0] for x1 in lst[1] for x2 in lst[2]]

它在这里返回正确的答案，但在我的情况下，我不知道嵌套了多少列表，lst所以如果列表的数量是可变的，我将如何有效地做到这一点？

标签： python