util.py 文件源码

def show_cluster(y, 
            yi, 
            c=['b', 'g', 'r', 'k'],
            title='Clustering Resutls'):
    #km = cluster.KMeans(n)
    #yi = km.fit_predict(y)
    #c = ['b', 'g', 'r', 'k']
    set_yi = list(set(yi))
    n = len(set_yi)
    for i in set_yi:
        sns.tsplot(y[yi==i], color=c[set_yi.index(i)])
    plt.title(title)