Python 每个列表一项的所有组合

问题描述

我有不同数量的不同长度的列表。我需要尝试为每个列表创建一个项目的所有可能性。这如何在 Python 中完成？

示例

l1 =  ['a','b','c']
l2 =  [1,2]
l3 =  [6]

# Expected result:
[
  ['a',1,6],['a',2,['b',['c',]

在这种情况下，我可以有 3 个循环，但由于这是针对数量不明确的列表，因此无法缩放。

我尝试使用 itertools，但不知道该用什么来处理我的案例，也在这里找不到类似的问题。因此，任何帮助将不胜感激。

解决方法

如果您有不同数量的列表，您应该使用列表列表，而不是单独的变量。然后您可以将列表扩展为 itertools.product() 的参数。

import itertools

lists = [
    ['a','b','c'],[1,2],[6]
]

print(list(itertools.product(*lists)))