Suppose I have a dataset with more than one columns. I groupby a certain column and partition the groupby dataset by some conditions. Now I have the index for every partitio