Simplify rpartition indexing