python绘制矩形树图

在Python中，可以使用matplotlib库中的`pyplot`模块来绘制矩形树状图，也称为层次条形图或分层柱状图。这是一种用于显示数据层级结构的有效工具，通常用来表示类别之间的大小差异。下面是一个简单的例子：

import matplotlib.pyplot as plt

# 假设我们有数据
data = {
    'A': {'B': 40, 'C': 55},
    'D': {'E': 60, 'F': 70},
}

# 计算总和
total = sum(sum(value.values()) for value in data.values())

# 创建图形并设置比例
fig, ax = plt.subplots(figsize=(8, 6))

# 画出每个级别
for key, sub_data in data.items():
    ax.barh(key, sub_data.values(), left=sum(data.values()) - sub_data.values())  # 向右调整累计值
    ax.text(-5, key, f'{sum(sub_data.values()):.0f}')

# 设置标题和轴标签，并调整边距
ax.set_title('Rectangle Tree Chart')
ax.set_xlabel('Value')
ax.set_xlim(0, total)
plt.tight_layout()

# 显示图形
plt.show()

在这个例子中，`barh()`函数用于水平绘制定义的高度的矩形，`text()`则添加了每个矩形的数值标签。你可以根据实际数据结构和需求修改这个示例。