Python numpy ndenumerate用法及代碼示例

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

用法:

class  numpy.ndenumerate(arr)

多維索引迭代器。

返回產生數組坐標和值對的迭代器。

參數：

arr： ndarray

輸入數組。

例子：

>>> a = np.array([[1, 2], [3, 4]])
>>> for index, x in np.ndenumerate(a):
...     print(index, x)
(0, 0) 1
(0, 1) 2
(1, 0) 3
(1, 1) 4