上一主题

numpy.nditer.reset

下一主题

numpy.ndindex.ndincr

numpy.ndindex

class numpy.ndindex(*shape)[source]

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

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

参数:

`* args`:ints

数组的每个维度的大小。

也可以看看

ndenumerateflatiter

例子

>>> 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() 标准迭代器方法,更新索引并返回索引元组。