使用python将分层节点的字典绘制为树状图

pip install networkx matplotlib

import networkx as nx
import matplotlib.pyplot as plt

def draw_tree(data, parent=None, tree=None, pos=None, layer=0):
    if tree is None:
        tree = nx.DiGraph()
        tree.add_node(parent)
        pos = {parent: (0, 0)}
    for child in data[parent]:
        tree.add_node(child)
        tree.add_edge(parent, child)
        pos[child] = (layer, -layer)
        if child in data:
            draw_tree(data, child, tree, pos, layer+1)
    return tree, pos

data = {
    'A': ['B', 'C'],
    'B': ['D', 'E'],
    'C': ['F', 'G'],
    'D': ['H'],
    'E': ['I', 'J'],
    'F': ['K'],
    'G': ['L'],
    'H': [],
    'I': [],
    'J': [],
    'K': [],
    'L': []
}

tree, pos = draw_tree(data)
nx.draw(tree, pos, with_labels=True, node_size=1000, node_color='lightblue', font_size=12, font_weight='bold')
plt.show()