partition_index function

Partition an index vector into an integer vector representing groups