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