SciPy

numpy.nditer.index

nditer.index