小编典典

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

python

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

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

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


阅读 238

收藏
2021-01-20

共1个答案

小编典典

您可以使用map映射的所有项目Drange,然后解开他们product

indices=list(itertools.product(*map(range, D)))
2021-01-20