kmeans++ python 对时间序列聚类/pandas列索引

最新推荐文章于 2024-07-14 23:05:10 发布

Jiiaaaoooo

最新推荐文章于 2024-07-14 23:05:10 发布

阅读量3.8k

点赞数

分类专栏： python

本文链接：https://blog.youkuaiyun.com/Jiiaaaoooo/article/details/89466436

版权

python 专栏收录该内容

16 篇文章

订阅专栏

estimator =KMeans(n_clusters=3,init = 'kmeans++')   #构造一个聚类数为5的聚类器,初始质心选取方式改为kmeans++
estimator.fit(data1)   #聚类
label_pred = estimator.labels_  #获取聚类标签
centroids = estimator.cluster_centers_ #获取聚类中心
print(label_pred)
#画图，每个类别画出一条线
for  i in range(len(label_pred)):
    if label_pred[i] == 0:
        x = [i for i in range(len(data1[1]))]
        plt.plot(x, data1[i], 'y')
    elif label_pred[i] == 1:
        x = [i for i in range(len(data1[1]))]
        plt.plot(x, data1[3], 'b')
    else:
        x = [i for i in range(len(data1[1]))]
        plt.plot(x, data1[i], 'r')

显示列索引
x = df.columns.values.tolist()