numpy.ndindex

class numpy.ndindex(*shape)

用于索引数组的N维迭代器对象。

给定数组的形状，ndindex实例遍历数组的N维索引。在每次迭代时，返回一个索引的元组，最后一个维度首先被迭代。

参数：

`* args`：ints

数组的每个维度的大小。

也可以看看

ndenumerate，flatiter

例子

>>> for index in np.ndindex(3, 2, 1):
...     print(index)
(0, 0, 0)
(0, 1, 0)
(1, 0, 0)
(1, 1, 0)
(2, 0, 0)
(2, 1, 0)

方法

ndincr()	将多维索引增加1。
next()	标准迭代器方法，更新索引并返回索引元组。