SciPy

numpy.nditer.iterindex

nditer.iterindex