Python numpy ndindex用法及代碼示例

本文簡要介紹 python 語言中 numpy.ndindex 的用法。

用法:

class  numpy.ndindex(*shape)

用於索引數組的 N 維迭代器對象。

給定數組的形狀，ndindex 實例會迭代數組的 N 維索引。每次迭代都會返回一個索引元組，首先迭代最後一個維度。

參數：

shape： 整數，或單個整數元組

數組每個維度的大小可以作為單獨的參數或作為元組的元素傳遞。

例子：

維度作為單獨的參數

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