random_split.py

import random
random.seed(777)
dataset = {}
for acc, members in acc_membership.items():
    if len(members) < 10 or len(members) > 1000:
        # too small or too big accounts
        continue
    group1 = {}
    group2 = {}
    for member in members:
        names = people[member]
        if 1 < len(names) < 20:
            first = random.sample(range(len(names)), len(names) // 2)
            second = set(range(len(names))).difference(first)
            names = list(names)
            group1[member] = [names[i] for i in first]
            group2[member] = [names[i] for i in second]
    if len(group1) > 1 and len(group2) > 2:
        dataset[str(acc)] = group1, group2