python: grouping based on frequency of occurrence

Question

I have labels and their frequencies(ie number of times they are repeated) for a dataset.

Is there a library which can be used to group together those labels which have almost similar frequency(ie based on variation).

As an example: Suppose a is repeated 10 times, b 9 times, c 6 times, d 5 times, e 2 times So I want and b fall into one group, c and d in one group and e in another group.

Answer 1

You can use the following function to group based upon count.

def group_labels(cnts): 
  d = {} 
  for k, v in cnts.items(): 
    d.setdefault(v, []).append(k)
  return sorted(d.values(), key=lambda x: x[0]) # sorted by first label

Example

cnts = {'a': 4, 'b': 15, 'c':4, 'd':16, 'e':1, 'f':16}
print(group_labels(cnts))
[['a', 'c'], ['b'], ['d', 'f'], ['e']]