python 如何看层次聚类树状图
时间: 2023-10-13 08:22:06 浏览: 154
chapter14_python聚类分析_python_antspu3_层次化聚类_
Python 中可以使用 matplotlib 和 scipy 库来绘制层次聚类树状图。以下是一个示例:
``` python
import matplotlib.pyplot as plt
from scipy.cluster.hierarchy import dendrogram, linkage
# 生成数据
X = [[1, 2], [1, 4], [1, 0], [4, 2], [4, 4], [4, 0]]
labels = range(1, 7)
# 计算距离矩阵和聚类
Z = linkage(X, 'ward')
# 绘制树状图
fig, ax = plt.subplots(figsize=(10, 5))
dendrogram(Z, labels=labels, ax=ax)
plt.show()
```
在上面的代码中,首先生成了一个二维数据集 X 和每个样本的标签 labels。然后使用 scipy 库中的 linkage 函数来计算距离矩阵和聚类,其中 'ward' 表示使用 Ward 方式进行聚类。最后使用 matplotlib 库中的 dendrogram 函数来绘制树状图,并指定标签。运行代码,会输出一个层次聚类树状图。
阅读全文