如何在python中创建层次树，从cs获取数据

Question

我试图通过从csv获取数据来创建一个层次结构树，如下所示：

Node 1  Node 2  Node 3
AL201   AL705   AL500
AL204   AL704   AL500
AL509   AL703   AL500
AL501   AL507   AL500
AL502   AL704   AL500
AL503   AL501   AL500
AL505   AL704   AL500
AL707   AL500   AL500
AL705   AL704   AL500

我已经写了这段代码，但是我没有得到我想要的，只有一个带有节点的图：

import pandas as pd
import networkx as nx
import matplotlib.pyplot as plt


data = pd.read_csv(r'C:\Users\flovascio\Documents\Python Scripts\Exercises\Test_Network_2.csv', delimiter = ';')



columns = list(data)
G = nx.DiGraph()
for index, row in data.iterrows():
    G.add_edge(row[0], row[1], length=150)
    G.add_edge(row[1], row[2], length=150)

nx.draw(G ,edge_color = 'blue' ,width=2, linewidths=1, node_size = 500, with_labels=True)
plt.draw()
plt.show()

有人能给我推荐另一种方法来得到一个层次树吗

谢谢