如何在python中绘制群集? [英] How to plot clusters in python?

问题描述

我正在使用 python sklearn.cluster 进行聚类.我有61个数据，每个数据的尺寸为26.原始数据:

I am using python sklearn.cluster to do clustering. I have 61 data and each data is of dimension 26. Original data:

UserID  Communication_dur   Lifestyle_dur   Music & Audio_dur   Others_dur  Personnalisation_dur    Phone_and_SMS_dur   Photography_dur Productivity_dur    Social_Media_dur    System_tools_dur    ... Music & Audio_Freq  Others_Freq Personnalisation_Freq   Phone_and_SMS_Freq  Photography_Freq    Productivity_Freq   Social_Media_Freq   System_tools_Freq   Video players & Editors_Freq    Weather_Freq
1   63  219 9   10  99  42  36  30  76  20  ... 2   1   11  5   3   3   9   1   4   8
2   9   0   0   6   78  0   32  4   15  3   ... 0   2   4   0   2   1   2   1   0   0


from sklearn.preprocessing import StandardScaler
from sklearn.decomposition import PCA 

Sc = StandardScaler()
X = Sc.fit_transform(df)

我已将PCA应用于数据框，以便基于K均值绘制聚类.

I have applied PCA to a dataframe in order to plot clusters based on K-means.

pca = PCA(3) 
pca.fit(X) 
pca_data = pd.DataFrame(pca.transform(X)) 
print(pca_data.head())

数据:

    0  1  2
 0  8 -4  5
 1 -2 -2  1
 2  1  1 -0
 3  2 -1  1
 4  3 -1 -3

kmeans_pca=KMeans(n_clusters=10,init="k-means++",random_state=42)
kmeans_pca.fit (pca_data)

现在我要绘制结果群集，我该怎么办?

Now I want to plot the resultant clusters how can i do ?

推荐答案

尚未测试，但可以使用以下代码进行可视化:

Haven't tested but can visualize with code like below:

import matplotlib.pyplot as plt
import seaborn as sns

def show_clusters(data, labels):
     palette = sns.color_palette('hls', n_colors=len(set(labels)))
     sns.scatterplot(x=data.iloc[:, 0], y=data.iloc[:, 1], hue=labels, palette=palette)
     plt.axis('off')
     plt.show()

然后通过传递PCA数据和K-means簇标签来调用该函数:

Then call the function by passing PCA data and K-means cluster labels:

show_clusters(pca_data, kmeans_pca.labels_)

输出:

这篇关于如何在python中绘制群集?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！