K-Means聚类算法并可视化
时间: 2023-06-29 16:13:20 浏览: 121
K-Means聚类算法是一种常见的无监督学习算法,可以将数据集分成K个类别。以下是一种K-Means聚类算法的可视化过程:
1. 首先,我们需要准备一个数据集。这里我们使用Python中的Scikit-learn库生成一个二维的随机数据集。
```python
from sklearn.datasets import make_blobs
import matplotlib.pyplot as plt
# 生成随机数据集
X, y = make_blobs(n_samples=100, centers=3, n_features=2, random_state=0)
# 绘制数据集
plt.scatter(X[:,0], X[:,1], s=50)
plt.show()
```
2. 然后,我们需要对数据集进行K-Means聚类。这里我们使用Python中的Scikit-learn库中的KMeans函数进行聚类。
```python
from sklearn.cluster import KMeans
# 进行聚类
kmeans = KMeans(n_clusters=3, random_state=0).fit(X)
```
3. 最后,我们需要将聚类结果可视化。这里我们使用Matplotlib库绘制各个聚类簇的中心点和数据点的聚类结果。
```python
# 绘制聚类结果
plt.scatter(X[:,0], X[:,1], c=kmeans.labels_, s=50)
plt.scatter(kmeans.cluster_centers_[:,0], kmeans.cluster_centers_[:,1], c='red', s=50)
plt.show()
```
以上就是K-Means聚类算法的可视化过程。
阅读全文