对多维索引（任意维度）的列表表示进行高效迭代

Question

我使用任意维度的多维结构。我有一个xrange迭代器的Python列表，每个迭代器代表一个多维数组的索引：

indices = [ i, j, k ]

，其中

i = xrange(1,3)
j = xrange(3,5)
k = xrange(5,7)

要生成所有可能的值，我使用以下天真递归代码：

def travtree(index,depth):
    "Recursion through index list"
    if depth >= len(indices):
        # Stopping Condition
        print index
    else:
        # Recursion
        currindexrange = indices[depth]
        for currindex in xrange(len(currindexrange)):
            newindex = list(index) # list copy
            newindex.append(currindexrange[currindex])
            travtree(newindex,depth+1)

travtree([],0)

这很好用，但我想知道，有更高效的Pythonic方法吗？我试着查看itertools模块，但没有任何事情发生在我身上。

Answer 1

>>> from itertools import product
>>> i = xrange(1,3)
>>> j = xrange(3,5)
>>> k = xrange(5,7)
>>> indices = [ i, j, k ]
>>> for item in product(*indices):
        print item


(1, 3, 5)
(1, 3, 6)
(1, 4, 5)
(1, 4, 6)
(2, 3, 5)
(2, 3, 6)
(2, 4, 5)
(2, 4, 6)