Python - n维立方体的角坐标

时间:2011-11-08 23:30:21

标签: python itertools hypercube

我正在尝试从每个维度的分钟和最大值列表中获取n维立方体的坐标。我能够使用for循环获得角落,但我想推广任意数量的维度。

例如:

mins = [-1,-2,-3]
maxes = [1,2,3]

会给出坐标:

(-1, -2, -3), (-1, -2, 3), (-1, 2, -3), (-1, 2, 3),
(1, 2, 3), (1, 2, -3), (1, -2, 3), (1, -2, -3)

这基本上是通过两个列表查找所有路径,从每个索引的列表中选择一个值。我已经看到算法给出路径数或最快路径,但是我没有找到一个列举所有可能路径的算法。

我认为itertools会进入解决方案,但无法弄清楚如何使用产品,排列和组合来提供所需的结果。最接近的是:

list(itertools.product((xmin, xmax), (ymin, ymax), (zmin, zmax)))

1 个答案:

答案 0 :(得分:5)

你非常接近,*zip( ... )正是你所寻找的:

>>> list(itertools.product(*zip([-1,-2,-3],[1,2,3])))
[(-1, -2, -3), (-1, -2, 3), (-1, 2, -3), (-1, 2, 3), (1, -2, -3), (1, -2, 3), (
, 2, -3), (1, 2, 3)]