Python 如何在点图中设置固定深度级别

我正在从一个树状的数据结构创建一个点图可视化，但是我很难根据数据类型设置固定的级别深度。例如，如果我在一棵树中有4个节点，a表示一种特定的数据类型，B表示另一种，那么它将像图1：

与图2相反：

图2是我想要的结果

我要找的是固定水平。我怎样才能做到这一点？我可以很容易地确定要添加到图中的数据类型，但在如何标记节点以实现这一点上遇到了困难。这可以用子图来完成吗

---

仅供参考，这是我第一次玩圆点。

是的，子图可以工作

digraph {
  subgraph { rank = same; A0 };
  subgraph { rank = same; B0; B1 };
  root -> A0;
  A0 -> B0;
  root -> B1;
}

导致

---

（来源：）

完美，这正是我想要的！比我想象的还要容易。谢谢

                                      ROOT   
                                     /    \  
                                   A[0]    \
                                   /        \
                                 B[0]       B[1]   

digraph {
  subgraph { rank = same; A0 };
  subgraph { rank = same; B0; B1 };
  root -> A0;
  A0 -> B0;
  root -> B1;
}