树状图绘制指南

在Python中，树状图可以使用networkx和matplotlib库绘制。下面我将展示一个完整的树状图绘制解决方案，包含中文标签和层次化布局。

import networkx as nx
import matplotlib.pyplot as plt
import matplotlib.font_manager as fm
import numpy as np

# 设置中文字体支持
plt.rcParams['font.sans-serif'] = ['SimHei']  # 用来正常显示中文标签
plt.rcParams['axes.unicode_minus'] = False    # 用来正常显示负号

# 创建一个有向图表示树结构
G = nx.DiGraph()

# 添加节点（示例：公司组织结构）
nodes = [
    ("CEO", "首席执行官"),
    ("CTO", "技术总监"),
    ("CFO", "财务总监"),
    ("HRD", "人力资源总监"),
    ("Dev", "开发团队"),
    ("QA", "质量保证团队"),
    ("Acct", "会计部门"),
    ("Rec", "招聘团队")
]

# 添加边（节点间关系）
edges = [
    ("CEO", "CTO"),
    ("CEO", "CFO"),
    ("CEO", "HRD"),
    ("CTO", "Dev"),
    ("CTO", "QA"),
    ("CFO", "Acct"),
    ("HRD", "Rec")
]

# 添加节点和边到图中
for node in nodes:
    G.add_node(node[0], label=node[1])

G.add_edges_from(edges)

# 定义层级布局
def hierarchy_pos(G, root, width=1., vert_gap=0.2, vert_loc=0, xcenter=0.5):
    pos = {}

    def h_recur(G, root, width=1., vert_gap=0.2, vert_loc=0, xcenter=0.5, 
                pos=None, parent=None, parsed=[]):
        if root not in parsed:
            parsed.append(root)
            neighbors = list(G.successors(root))

            if len(neighbors) != 0:
                dx = width / len(neighbors)
                nextx = xcenter - width/2 - dx/2

                for neighbor in neighbors:
                    nextx += dx
                    pos = h_recur(G, neighbor, width=dx, vert_gap=vert_gap, 
                                 vert_loc=vert_loc-vert_gap, xcenter=nextx,
                                 pos=pos, parent=root, parsed=parsed)
                    pos[neighbor] = (nextx, vert_loc)

            if parent is not None:
                pos[root] = (xcenter, vert_loc)
            else:
                pos[root] = (xcenter, vert_loc)

        return pos

    return h_recur(G, root, width, vert_gap, vert_loc, xcenter, pos={}, 
                  parent=None, parsed=[])

# 使用自定义布局
pos = hierarchy_pos(G, "CEO")

# 创建图形
plt.figure(figsize=(12, 8))

# 绘制节点
node_labels = nx.get_node_attributes(G, 'label')
nx.draw_networkx_nodes(G, pos, node_size=3000, node_color='skyblue', alpha=0.9)
nx.draw_networkx_edges(G, pos, width=2, edge_color='gray', arrows=True, 
                       arrowstyle='-|>', arrowsize=20)

# 绘制中文标签
nx.draw_networkx_labels(G, pos, labels=node_labels, font_size=12, 
                        font_family='SimHei')

# 添加标题和说明
plt.title("公司组织结构树状图", fontsize=16, pad=20)
plt.figtext(0.5, 0.01, 
            "树状图展示层级关系：节点大小表示重要性，箭头表示从属关系", 
            ha="center", fontsize=12, style='italic')

# 添加网格和美化
plt.grid(False)
plt.axis('off')

# 调整布局
plt.tight_layout()
plt.subplots_adjust(bottom=0.15)

# 显示图形
plt.show()

代码说明

这个树状图绘制代码包含以下关键部分：

1. 中文支持：

   • 使用plt.rcParams设置中文字体（SimHei）
   • 禁用unicode负号显示以解决兼容性问题

2. 树结构定义：

   • 使用networkx.DiGraph()创建有向图
   • 添加节点（节点ID和中文标签）
   • 添加边定义节点间关系

3. 自定义层级布局：

   • hierarchy_pos函数实现树状布局算法
   • 根节点在顶部，子节点均匀分布在下方

4. 可视化元素：

   • 节点使用天蓝色圆形表示
   • 箭头表示层级关系方向
   • 中文标签清晰显示
   • 添加标题和说明文字

5. 美化设置：

   • 适当大小和间距
   • 禁用坐标轴显示
   • 添加网格背景

这段代码可以直接运行，生成一个美观的公司组织结构树状图。您可以根据需要修改节点和边的定义来创建自己的树状结构。

如何自定义树状图

1. 修改节点：

   • 在nodes列表中添加/删除元组，格式为(节点ID, 显示文本)

2. 修改关系：

   • 在edges列表中添加/删除连接关系，格式为(上级节点ID, 下级节点ID)

3. 样式调整：

   • 修改node_size调整节点大小
   • 修改node_color改变节点颜色
   • 调整figsize参数改变图像尺寸
   • 修改font_size调整文字大小

希望这个示例能帮助您创建所需的树状图！