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

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()