python绘制矩形树状图

在Python中，我们可以使用matplotlib库来绘制矩形树状图，也称为层次聚类图或Dendrogram。这种图表常用于可视化数据集的相似度矩阵，特别是在统计学和机器学习中，尤其是在对数据进行聚类分析时。

下面是一个简单的例子，展示如何使用scipy和matplotlib创建一个矩形树状图：

import numpy as np
from scipy.cluster.hierarchy import dendrogram, linkage
import matplotlib.pyplot as plt

# 假设你有一个二维数组data作为样本数据
data = np.random.rand(5, 4)

# 对数据进行单链加权平均的聚类
Z = linkage(data, 'ward')

# 使用dendrogram函数绘制图形
plt.figure(figsize=(10, 6))
dendrogram(Z,
           leaf_font_size=8,  # 设置节点标签的字体大小
           labels=data.index.tolist(),  # 设置节点标签
           orientation='left')  # 可选择'right', 'top'

plt.title('Hierarchical Clustering Dendrogram')
plt.xlabel('Sample Index')
plt.ylabel('Distance')
plt.show()

在这个例子中，`linkage`函数生成了一个聚类链接矩阵，`dendrogram`函数则将其转换成可视化的图形。你可以通过调整参数来自定义颜色、线型等样式。