numpy.ndindex

类 numpy. ndindex ( *形状)

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

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

参数：

形状整数，或单个整数元组

数组每个维度的大小可以作为单独的参数或作为元组的元素传递。

也可以看看

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)

相同的尺寸 - 但在一个元组中(3, 2, 1)

>>> 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()

将多维索引加一。