cluster.py

cluster_data.py

from collections import defaultdict

def cluster_data(lst_array, max_range):
    clustered = defaultdict(list)
    n = 1
    for lst in lst_array:
        clustered[n].append(lst)
        n = n+1
        if n > max_range:
            n = 1
    return clustered