numpy.ndindex

class numpy.ndindex(*shape)

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

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

参数:

shape整数,或一个整数元组

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

参见

ndenumerate , flatiter

示例

Try it in your browser!

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

Go BackOpen In Tab

方法

ndincr ()

将多维索引递增1.