使用k-means聚类后并用折线图的形式显示
时间: 2023-06-26 18:07:21 浏览: 150
好的,我可以为你演示如何使用K-means算法聚类,并用折线图展示。以下是一个Python示例代码,使用sklearn库进行聚类和matplotlib库绘制折线图:
```python
import numpy as np
from sklearn.cluster import KMeans
import matplotlib.pyplot as plt
# 创建数据集
X = np.array([[1, 2], [1, 4], [1, 0], [4, 2], [4, 4], [4, 0]])
# 使用KMeans算法进行聚类
kmeans = KMeans(n_clusters=2, random_state=0).fit(X)
# 获取聚类结果
labels = kmeans.labels_
centers = kmeans.cluster_centers_
# 绘制折线图
plt.figure(figsize=(8, 6))
colors = ['r', 'g', 'b', 'y', 'c', 'm']
for i in range(len(X)):
plt.plot([X[i][0], centers[labels[i]][0]], [X[i][1], centers[labels[i]][1]], colors[labels[i]])
plt.scatter(centers[:, 0], centers[:, 1], marker='x', s=200, linewidths=3, color='k')
plt.show()
```
运行代码后,会得到一个带有聚类结果和聚类中心的折线图,如下所示:
![K-means折线图示例](https://i.imgur.com/woQqk9a.png)
在这个示例中,我们使用K-means算法将数据集分为两类,用红色和绿色的折线表示两个聚类。黑色的叉标表示聚类中心。
阅读全文