util.py 文件源码

def show_km(y, 
            n=4, 
            c=['b', 'g', 'r', 'k'],
            title='KMeans Clustering'):
    km = cluster.KMeans(n)
    yi = km.fit_predict(y)
    #c = ['b', 'g', 'r', 'k']
    for i in range(n):
        sns.tsplot(y[yi==i], color=c[i])
    plt.title(title)