kmeans聚类算法python代码可视化
时间: 2023-03-29 09:02:18 浏览: 129
python实现kmeans聚类算法
5星 · 资源好评率100%
可以使用matplotlib库来可视化kmeans聚类算法的python代码。具体实现方法可以参考以下代码:
```python
import numpy as np
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans
# 生成数据
X = np.random.rand(100, 2)
# 聚类
kmeans = KMeans(n_clusters=3, random_state=).fit(X)
# 可视化
plt.scatter(X[:, ], X[:, 1], c=kmeans.labels_)
plt.scatter(kmeans.cluster_centers_[:, ], kmeans.cluster_centers_[:, 1], marker='x', s=200, linewidths=3, color='r')
plt.show()
```
这段代码可以生成100个随机二维数据,然后使用kmeans算法将其聚成3类,并用散点图可视化结果。其中,`kmeans.labels_`表示每个数据点所属的类别,`kmeans.cluster_centers_`表示每个类别的中心点。
阅读全文