如何在networkx中画出更漂亮的树

import networkx as nx
import matplotlib.pyplot as plt

G = nx.Graph()  # 创建无向图
G = nx.DiGraph()  # 创建有向图

G.add_node("A")  # 添加节点A
G.add_edge("A", "B")  # 添加边A-B

node_style = {
    "node_color": "lightblue",  # 节点颜色
    "node_size": 500,  # 节点大小
    "node_shape": "o",  # 节点形状
    "alpha": 0.8,  # 节点透明度
    "font_size": 12,  # 节点标签字体大小
    "font_color": "black",  # 节点标签字体颜色
    "font_weight": "bold",  # 节点标签字体粗细
    "edge_color": "gray",  # 边颜色
    "width": 1,  # 边宽度
    "style": "solid"  # 边线样式
}

pos = nx.spring_layout(G)  # 设置节点布局算法
nx.draw(G, pos, with_labels=True, **node_style)  # 绘制图形

plt.axis("off")  # 关闭坐标轴
plt.show()  # 显示图形