在matplotlib中绘制二叉树

import matplotlib.pyplot as plt
import networkx as nx

G = nx.DiGraph()

nodes = ['A', 'B', 'C', 'D', 'E', 'F', 'G']
edges = [('A', 'B'), ('A', 'C'), ('B', 'D'), ('B', 'E'), ('C', 'F'), ('C', 'G')]

G.add_nodes_from(nodes)
G.add_edges_from(edges)

pos = nx.nx_pydot.graphviz_layout(G, prog='dot')

plt.figure(figsize=(8, 6))
nx.draw_networkx(G, pos, with_labels=True, node_size=500, node_color='lightblue', font_size=12, font_color='black', edge_color='gray', width=1, arrows=True)
plt.axis('off')
plt.show()