写一个矩形树图的案例python

下面是一个简单的矩形树图案例，使用Python和matplotlib绘制。

import matplotlib.pyplot as plt
import numpy as np

# 定义数据
data = {
    'A': {
        'B': {
            'E': None,
            'F': None
        },
        'C': {
            'G': None,
            'H': None
        },
        'D': None
    }
}

# 定义绘图函数
def plot_rect_tree(data, x0, y0, dx, dy):
    if not data:
        return
    
    # 计算当前节点的中心坐标
    x = x0 + dx / 2
    y = y0 + dy / 2
    
    # 绘制当前节点的矩形
    plt.gca().add_patch(plt.Rectangle((x - dx / 2, y - dy / 2), dx, dy, fill=False))
    
    # 绘制当前节点的文本
    plt.text(x, y, list(data.keys())[0], ha='center', va='center')
    
    # 绘制子节点
    if isinstance(list(data.values())[0], dict):
        num_children = len(list(data.values())[0])
        dx_child = dx / num_children
        x_child = x0 + dx_child / 2
        y_child = y0 - dy / 2
        
        for child_data in list(data.values())[0].values():
            plot_rect_tree(child_data, x_child, y_child, dx_child, dy)
            x_child += dx_child

# 设置图形参数
fig, ax = plt.subplots(figsize=(8, 6))
ax.set_xlim(0, 8)
ax.set_ylim(0, 6)
ax.set_aspect('equal')
plt.axis('off')
plt.title('Rectangular Tree')

# 绘制矩形树图
plot_rect_tree(data, 1, 5, 6, 2)

# 显示图形
plt.show()

这个案例只是一个简单的示例，可以根据具体需求进行修改和扩展。