使用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))]