groups.py

from math import floor 

def assign_equal_groups(sample_size, num_groups):

  return_list = [];
  group_size = sample_size / num_groups;

  for i in range(0, sample_size):
      v = (i / group_size) % num_groups;
      return_list.append(floor(v));

  return return_list;
  

print(assign_equal_groups(50, 10));