Python画图展示层级关系

在 Python 中，使用 Matplotlib 或 NetworkX 库可以有效地展示层级关系，例如树状结构或组织架构图。以下是两种常见方法的简要介绍：

1. **Matplotlib**（用于绘制简单的数据可视化）:
- **Hierarchical Layouts**: Matplotlib 的 `pyplot` 模块虽然主要用于线图和散点图，但它通过 `plt.sankey()` 或 `plt.subplot_mosaic()` 函数可以用来制作一些简单的层级结构图，如流程图或资源分配图。

   import matplotlib.pyplot as plt
   nodes = ['A', 'B', 'C', 'D', 'E']
   links = [('A', 'B'), ('B', 'C'), ('C', 'D'), ('D', 'E')]
   G = nx.DiGraph(links)
   pos = nx.nx_pydot.graphviz_layout(G, prog='dot')
   nx.draw(G, pos, with_labels=True)
   plt.show()

2. **NetworkX**（专为图形分析设计）:
- **Tree and Hierarchical Graphs**: NetworkX 提供了专门用于处理树形结构和层级关系的功能。你可以创建一个 `nx.DiGraph` 对象，并使用 `draw_tree()` 函数或者 `pos = nx.spring_layout(G)` 后再用 `nx.draw_networkx_nodes()` 和 `nx.draw_networkx_edges()` 分别绘制节点和边。

   import networkx as nx
   G = nx.DiGraph()
   G.add_nodes_from(nodes)
   G.add_edges_from(links)
   pos = nx.spring_layout(G)  # 使用布局算法
   nx.draw(G, pos, with_labels=True, node_color='lightblue', edge_color='gray', font_weight='bold')
   plt.show()