用python实现聚类
时间: 2023-10-30 13:08:45 浏览: 61
聚类python
下面是一个简单的用Python实现聚类的例子:
```python
import numpy as np
from sklearn.cluster import KMeans
import matplotlib.pyplot as plt
# 生成数据
X = np.random.rand(100, 2)
# 聚类
kmeans = KMeans(n_clusters=3, random_state=0).fit(X)
# 绘制结果
plt.scatter(X[:, 0], X[:, 1], c=kmeans.labels_)
plt.show()
```
该代码使用了scikit-learn库中的KMeans算法进行聚类。首先生成了一个包含100个随机点的二维数据集,然后使用KMeans算法将这些点分为3类,并将结果绘制在散点图上。可以看到,不同的类别被用不同的颜色表示。
除了KMeans算法,scikit-learn库中还包含了许多其他的聚类算法,如层次聚类、DBSCAN等,可以根据具体需求选择合适的算法。
阅读全文