能否提供一个Python代码示例,展示如何使用层次聚类(如Agglomerative Clustering)算法来进行数据聚类?
时间: 2024-10-29 19:09:10 浏览: 10
Python实现简单层次聚类算法以及可视化
当然可以。这里是一个简单的Python代码示例,使用`scikit-learn`库中的层次聚类模块`AgglomerativeClustering`进行数据聚类:
```python
# 导入所需库
import numpy as np
from sklearn.cluster import AgglomerativeClustering
import matplotlib.pyplot as plt
# 假设我们有一个二维数据集
data = np.random.rand(100, 2) # 创建一个随机生成的数据集
# 定义聚类层数 (n_clusters),这里设置为3
n_clusters = 3
# 实例化AgglomerativeClustering模型
agglom = AgglomerativeClustering(n_clusters=n_clusters)
# 训练模型
clusters = agglom.fit_predict(data)
# 绘制数据点及其所属簇
plt.scatter(data[:, 0], data[:, 1], c=clusters)
plt.xlabel('Feature 1')
plt.ylabel('Feature 2')
plt.title('Hierarchical Clustering Result')
plt.show()
#
阅读全文