Как создать иерархическое дерево в Python, получая данные из CSV - PullRequest
Новая
       60

Как создать иерархическое дерево в Python, получая данные из CSV

0 голосов
/ 02 июля 2019

Я пытаюсь создать иерархическое дерево, получая данные из 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

Я написал этот кусок кода, но я не получил именно то, что хотел, имея только граф с узлами: graph

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()

Может кто-нибудь предложить мне другой способ получить иерархическое дерево, пожалуйста?

Спасибо

...