mathutil.py 文件源码

def constructParallelograms(dataset):
    '''
    @params
        dataset is a list of points to find clusters in
    returns a list of the parallelograms found.
    '''
    af = AffinityPropagation().fit(dataset)
    print(af.cluster_centers_, af.labels_, len(af.cluster_centers_))
    clusters = []
    count = 0
    while (count < len(af.cluster_centers_)):
        pointlist = af.cluster_centers_[count].tolist()
        clusters += [Point(pointlist[0], pointlist[1])]
        count += 1

    print(clusters)
    return extrapolateParallelogram(clusters[0], clusters[1], clusters[2])