如何解析 Python 中的樹？

Question

我需要幫助來開發我正在研究的這個算法。 我有以下格式的樹的輸入：

Root -> AB CD
AB -> ABC CBA
CD -> CDE FGH

該算法應該讀取括號格式並給出以下 output：

                    Root
                     |
                ____________
              AB           CD
              |             |  
       __________         ___________
      ABC      CBA        CDE      FGH

Answer 1

也許你可以從networkx模塊開始

1. 安裝網絡x： pip pip install networkx

2. 安裝matplotlib ： pip install matplotlib

3. 程序

import networkx as nx
import matplotlib.pyplot as plt

if __name__ == "__main__":
    # define DiGraph
    tree = nx.DiGraph()

    # add node
    tree.add_node("root")
    tree.add_node("AB")
    tree.add_node("CD")
    tree.add_node("ABC")
    tree.add_node("CBA")
    tree.add_node("CDE")
    tree.add_node("FGH")

    # add connection edge
    tree.add_edge("root","AB")
    tree.add_edge("root","CD")
    tree.add_edge("AB","ABC")
    tree.add_edge("AB","CBA")
    tree.add_edge("CD","CDE")
    tree.add_edge("CD","FGH")

    # output the tree
    nx.draw(tree,with_labels = True)
    plt.show()