# 导入必要的库
import numpy as np
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)
# 初始化KMeans模型
kmeans = KMeans(n_clusters=4)
# 拟合模型
kmeans.fit(X)
# 预测聚类标签
labels = kmeans.predict(X)
# 获取聚类中心
centroids = kmeans.cluster_centers_
# 可视化聚类结果
plt.scatter(X[:, 0], X[:, 1], c=labels, cmap='viridis')
plt.scatter(centroids[:, 0], centroids[:, 1], marker='*', s=300, c='red')
plt.title('KMeans Clustering')
plt.show()
numpy用于数值计算,sklearn.cluster.KMeans用于KMeans聚类算法,sklearn.datasets.make_blobs用于生成模拟数据,以及matplotlib.pyplot用于绘图。make_blobs函数生成具有4个簇的300个样本点。上一篇:python定义列表
下一篇:python file怎么打开
Laravel PHP 深圳智简公司。版权所有©2023-2043 LaravelPHP 粤ICP备2021048745号-3
Laravel 中文站