python - 如何从 python 中的二维列表中获取所有可能的项目组合？

Question

我没有找到更好的方式来在标题中表达这个问题。如果可以，请编辑。

我有一个这样的列表列表:

a = [['a','b'],[1,2]]

现在，我想要一个像这样吐出所有可能组合的函数:

[['a',1],['a',2],['b',1],['b',2]]

其中 a 中的列表数量既不是预先知道的，也不是预先知道每个子列表的长度的，但是所有出现的组合都应该包含每个子列表中的 1 个项目。

Answer 1

你需要itertools.product() :

>>> list(itertools.product(*a))
[('a', 1), ('a', 2), ('b', 1), ('b', 2)]