numpy.select(condlist, choicelist, default=0)

class numpy.flatiter

nditer.iternext() Check whether iterations are left, and perform a single internal iteration without returning the result

numpy.compress(condition, a, axis=None, out=None)

ndindex.next()

flatiter.copy() Get a copy of the iterator as a 1-D array. Examples