Hierarchical Clustering from distance function

scipy_dist_clustering.py

def cluster_from_distances(data_points, dist_func, cluster_num):
    from scipy.cluster.hierarchy import linkage, cut_tree
    points = list()
    for d1 in data_points:
        d_list = [dist_func(d1,d2) for d2 in data_points]
        points.append(d_list)
    import scipy.spatial.distance as ssd
    points = ssd.squareform(points)

    Z = linkage(points, 'ward')
    cluster_indicators = cut_tree(Z, n_clusters=cluster_num)
    cluster_indicators = [c[0] for c in cluster_indicators]
    return cluster_indicators