def indices(shape, type=None): """indices(shape, type=None) returns an array representing a grid of indices with row-only, and column-only variation. """ shape = tuple(shape) a = concatenate(ufunc.nonzero(_nc.ones(shape))) a.setshape((len(shape),)+shape) if type is not None: a = a.astype(type) return a
def _compress(condition, a): return _take(a, ufunc.nonzero(condition))