使用Python中的scikit-learn库来实现k均值聚类模型
时间: 2024-02-21 07:58:33 浏览: 80
好的,下面是一个完整的Python代码示例,使用scikit-learn库来实现k均值聚类模型:
```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.60, random_state=0)
# 创建k均值聚类模型
kmeans = KMeans(n_clusters=4)
# 拟合模型
kmeans.fit(X)
# 预测每个样本所属的簇
y_kmeans = kmeans.predict(X)
# 绘制聚类结果
plt.scatter(X[:, 0], X[:, 1], c=y_kmeans, s=50, cmap='viridis')
centers = kmeans.cluster_centers_
plt.scatter(centers[:, 0], centers[:, 1], c='black', s=200, alpha=0.5)
plt.show()
```
这个代码示例创建了一个模拟数据集,并使用KMeans类创建了一个k均值聚类模型。然后,模型被拟合到数据集上,并使用predict()函数预测每个样本所属的簇。最后,使用matplotlib库绘制了聚类结果。
你可以根据需要对代码进行修改,例如改变数据集或者调整模型参数。希望能对你有所帮助!
阅读全文