聚类算法kmeans用python实现
时间: 2023-11-03 22:06:50 浏览: 94
Kmeans均值聚类算法原理以及Python如何实现
Kmeans是一种常用的聚类算法,可以用Python进行实现。您可以使用sklearn库中的KMeans类来实现Kmeans聚类算法。首先,导入sklearn库和数据集,然后创建一个KMeans对象并设置聚类的数量,最后使用fit方法拟合数据集。下面是一个简单的代码示例:
```
from sklearn.cluster import KMeans
# 导入数据集
data = [[x1, y1], [x2, y2], ...]
# 创建KMeans对象并设置聚类数量
kmeans = KMeans(n_clusters=3)
# 拟合数据集
kmeans.fit(data)
# 获取聚类结果
labels = kmeans.labels_
# 获取聚类中心
centers = kmeans.cluster_centers_
```
阅读全文