使用HDBSCAN对2维数据进行聚类
时间: 2024-02-04 11:04:29 浏览: 55
聚类二维数据
好的,以下是一个使用HDBSCAN对2维数据进行聚类的示例代码:
``` python
import numpy as np
import matplotlib.pyplot as plt
import hdbscan
# 生成2维数据
np.random.seed(0)
data = np.random.randn(100, 2)
# 使用HDBSCAN进行聚类
clusterer = hdbscan.HDBSCAN(min_cluster_size=5)
clusterer.fit(data)
# 可视化聚类结果
color_palette = plt.cm.get_cmap('Paired', np.max(clusterer.labels_) + 1)
cluster_colors = [color_palette(x) if x >= 0 else (0.5, 0.5, 0.5) for x in clusterer.labels_]
plt.scatter(data[:, 0], data[:, 1], s=50, linewidth=0, c=cluster_colors, alpha=0.5)
plt.show()
```
在这个示例中,我们生成了100个二维数据点,并使用HDBSCAN算法对其进行聚类。在可视化聚类结果时,我们使用了一个颜色映射来将不同的聚类标签映射到不同的颜色。
阅读全文