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

將多維索引遞增一。