Grouping or partitioning data by key values