ndindex

Top

Questions to David Rotermund

numpy.ndindex

class numpy.ndindex(*shape)[source]

An N-dimensional iterator object to index arrays.

Given the shape of an array, an ndindex instance iterates over the N-dimensional index of the array. At each iteration a tuple of indices is returned, the last dimension is iterated over first.

import numpy as np

for index in np.ndindex((3, 2, 1)):
    print(index)

Output:

(0, 0, 0)
(0, 1, 0)
(1, 0, 0)
(1, 1, 0)
(2, 0, 0)
(2, 1, 0)

The source code is Open Source and can be found on GitHub.