Seeded Groups

seeded_groups.py

def seeded_groups(lst, n):
    l = list(lst)

    # create empty groups
    groups = [[] for i in range(n)]

    # prefill the groups with placeholders
    for idx, elem in enumerate(l):
        mod_idx = idx % n
        groups[mod_idx].append(0)

    # replace placeholders
    for group in groups:
        for idx, elem in enumerate(group):
            group[idx] = l.pop(0)

    return groups

print("Seeded\n%r" % seeded_groups(range(1, 11), 3))