Improve indexing options with numpy.ix_