group the same key before shuffle, reduce the comparison during sorting