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

Question

I'm trying to solve the problem mentioned in this post . Consider the D=[d1,...,dm] a list of non-negative integers. I want to have the set of the Cartesian products of range(d1),...,range(dm) . For example if m=3 I could use itertools:

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

I would appreciate if you could help me know how I can generate the indices using D with arbitrary length.

Answer 1

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

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

Answer 2

您可以使用* ：

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