numpy.ndenumerate — NumPy v1.11 Manual (original) (raw)

class numpy.ndenumerate(arr)[source]

Multidimensional index iterator.

Return an iterator yielding pairs of array coordinates and values.

Parameters: arr : ndarray Input array.

Examples

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

Methods

next() Standard iterator method, returns the index tuple and array value.