sklearn kmeans 实例
时间: 2023-07-12 10:46:11 浏览: 86
KMeans简单示例
4星 · 用户满意度95%
好的,下面是一个使用sklearn实现KMeans聚类的简单示例:
```python
from sklearn.cluster import KMeans
from sklearn.datasets import make_blobs
import matplotlib.pyplot as plt
# 生成随机数据
X, y = make_blobs(n_samples=300, centers=4, cluster_std=0.6, random_state=0)
# 构建KMeans模型
kmeans = KMeans(n_clusters=4)
# 拟合数据并进行聚类
kmeans.fit(X)
# 预测聚类结果
y_pred = kmeans.predict(X)
# 绘制聚类结果
plt.scatter(X[:, 0], X[:, 1], c=y_pred)
plt.title("KMeans Clustering")
plt.show()
```
这个示例生成了一个包含4个聚类中心的随机数据集,并使用KMeans算法进行聚类。聚类结果通过散点图进行展示。
阅读全文