a portable and compact version of numpy.ndindex

ndindex.py

import functools, operator
prod = functools.partial(functools.reduce, operator.mul)
def ndindex(maxvals):
    total = prod(maxvals)
    index = [0] * len(maxvals)
    for _ in range(total):
        yield tuple(index)
        for i, n in enumerate(reversed(maxvals)):
            index[-i-1] += 1
            if index[-i-1] < n:
                break
            else:
                index[-i-1] = 0
     raise StopIteration