numpy.ndindex

class numpy.ndindex(*shape)

一个用于索引数组的 N 维迭代器对象.

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

参数:

shape整数,或一个整数的单个元组

数组的每个维度的尺寸可以作为单独的参数传递,也可以作为元组的元素传递.

参见

ndenumerate, flatiter

示例

>>> import numpy as np

作为单独参数的维度

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

将多维索引增加一.