编写python代码实现kmeans聚类算法
时间: 2023-09-10 08:14:16 浏览: 116
大数据的作业 基于python实现kmeans 是对球队强弱进行聚类+源代码+文档说明+数据
5星 · 资源好评率100%
好的,关于kmeans聚类算法的Python代码实现,可以使用sklearn库中的KMeans函数来实现。下面给出一个简单的示例代码:
```python
from sklearn.cluster import KMeans
import numpy as np
# 构造数据集
X = np.array([[1, 2], [1, 4], [1, 0],
[4, 2], [4, 4], [4, 0]])
# 定义聚类个数
kmeans = KMeans(n_clusters=2, random_state=0).fit(X)
# 输出各个数据点所属的簇
print(kmeans.labels_)
```
这段代码构造了一个二维数据集,然后使用KMeans函数将其分成了两个簇。输出的结果为每个数据点所属的簇的标签。
阅读全文