Group the People Given the Group Size They Belong To LeetCode solution in Python

1
# Runtime: 75 ms (Top 97.76%) | Memory: 14 MB (Top 88.55%)
2
class Solution(object):
3
    def groupThePeople(self, groupSizes):
4
        """
5
        :type groupSizes: List[int]
6
        :rtype: List[List[int]]
7
        """
8
        dict_group = {}
9
        for i in range(len(groupSizes)):
10
            if groupSizes[i] not in dict_group:
11
                dict_group[groupSizes[i]] = [i]
12
            else:
13
                dict_group[groupSizes[i]].append(i)
14
        return_list = []
15
        for i in dict_group:
16
            num_list = dict_group[i]
17
            for j in range(0, len(num_list), i):
18
                return_list.append(num_list[j : j + i])
19
        return return_list