k-means聚类算法python代码
时间: 2023-03-29 12:04:00 浏览: 125
K-means聚类python代码
以下是 k-means 聚类算法的 Python 代码:
```python
import numpy as np
from sklearn.cluster import KMeans
# 生成随机数据
data = np.random.rand(100, 2)
# 聚类
kmeans = KMeans(n_clusters=3)
kmeans.fit(data)
# 输出聚类结果
print(kmeans.labels_)
```
这段代码可以生成 100 个二维随机数据,并使用 k-means 算法将其聚为 3 类。最后输出每个数据点所属的类别。
阅读全文