在k-meams聚类中将不同的5个簇的样本点用不同颜色显示,用特殊标记显示簇中心,完成聚类结果的可视化
时间: 2024-03-22 09:37:28 浏览: 143
好的,这是一个关于k-means聚类可视化的问题。
首先,我们需要使用Python中的机器学习库来执行k-means聚类。这里我将使用scikit-learn库。
以下是一些代码,可以生成聚类结果的可视化图形:
```python
from sklearn.cluster import KMeans
import matplotlib.pyplot as plt
import numpy as np
# 创建一个数据集,包含 100 个随机点
X = np.random.rand(100, 2)
# 执行 k-means 聚类,将数据集分成 5 个簇
kmeans = KMeans(n_clusters=5).fit(X)
# 获取每个簇的中心点
centers = kmeans.cluster_centers_
# 获取每个数据点所属的簇
labels = kmeans.labels_
# 将不同簇的样本点用不同颜色显示
colors = ['red', 'green', 'blue', 'yellow', 'purple']
plt.scatter(X[:,0], X[:,1], c=[colors[label] for label in labels])
# 用特殊标记显示簇中心
markers = ['x', 'o', 's', 'D', '+']
for i, center in enumerate(centers):
plt.scatter(center[0], center[1], c=colors[i], marker=markers[i], s=200)
plt.show()
```
在这个例子中,我们生成了一个包含100个随机点的数据集,并将其分成5个簇。我们使用不同的颜色来显示每个簇的样本点,并使用特殊标记显示每个簇的中心点。
你可以根据需要修改代码中的数据集和簇数,并更改颜色和标记,以生成不同的聚类结果可视化图形。
阅读全文