KMeans 聚类结果可视化:使用散点图展示聚类结果
可以使用 matplotlib 库绘制散点图来显示 kmeans 聚类的结果。
首先,需要将用户数据降维至二维空间,可以使用主成分分析(PCA)或 t-SNE 等降维方法。然后,根据聚类结果,将不同类别的用户用不同的颜色或标记绘制在散点图上。
以下是一个示例代码:
import matplotlib.pyplot as plt
from sklearn.decomposition import PCA
# 将用户数据降至二维空间
pca = PCA(n_components=2)
user_data_2d = pca.fit_transform(user_data)
# 绘制散点图
colors = ['r', 'g', 'b', 'y']
markers = ['o', 's', '^', 'd']
for i in range(len(user_km.cluster_centers_)):
plt.scatter(user_data_2d[user_km.labels_ == i, 0],
user_data_2d[user_km.labels_ == i, 1],
c=colors[i], marker=markers[i], label='Cluster ' + str(i))
plt.scatter(user_centers.iloc[:, 0], user_centers.iloc[:, 1],
c='k', marker='*', s=150, label='Cluster Centers')
plt.legend()
plt.show()
其中,user_km.labels_ 表示每个用户被分配到的聚类标签,user_centers.iloc[:, 0] 和 user_centers.iloc[:, 1] 分别表示聚类中心在降维后的二维空间中的坐标。可以根据具体情况修改颜色、标记和标签等参数。
原文地址: https://www.cveoy.top/t/topic/oRk0 著作权归作者所有。请勿转载和采集!