使用itertools生成列表列表的笛卡尔积

Question

我试图解决中提到的问题这篇文章 。 考虑D=[d1,...,dm]非负整数的列表。 我想要一组range(d1),...,range(dm)的笛卡尔积。 例如，如果m=3我可以使用itertools：

 indices=[i for i in itertools.product(range(d1),range(d2),range(d3))]

如果您能帮助我知道如何使用任意长度的D生成indices将不胜感激。

Answer 1

您可以使用map将D所有项目映射到range ，然后将它们解压为product ：

indices=list(itertools.product(*map(range, D)))

Answer 2

您可以使用* ：

[i for i in itertools.product(*map(range, D))]