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.