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)

返回在新标签页中打开