Как мне получить DOT для вывода своего узла?

Question

Я использую pydotplus для анализа файла точек. Функция graph = pydotplus.graph_from_dot_file(dot_file) возвращает объект pydotplus.graphviz.Dot, я хочу напечатать узлы графа, информацию о ребрах, сохранить и использовать содержимое в моей программе на python. Кажется, DOT - особый класс, я не могу напечатать его содержимое. Как получить контент (узлы, ребра), чтобы я мог сохранить и использовать его в моей структуре данных с python?

Answer 1

Класс Dot наследуется от класса Graph, который имеет методы get_nodes () и get_edges (). Вот пример получения данных по узлам и ребрам.

import pydotplus
from sklearn import tree

# Data Collection
X = [[180, 15, 0],
     [177, 42, 0],
     [136, 35, 1],
     [174, 65, 0],
     [141, 28, 1]]

Y = ['1', '2', '1', '2', '1']

data_feature_names = ['a', 'b', 'c']

# Training
clf = tree.DecisionTreeClassifier()
clf = clf.fit(X, Y)

dot_data = tree.export_graphviz(clf,
                                feature_names=data_feature_names,
                                out_file=None,
                                filled=True,
                                rounded=True)
graph = pydotplus.graph_from_dot_data(dot_data)
print(graph.to_string())

for node in graph.get_nodes():
    print(node.get_name())
    print(node.get_port())
    print(node.to_string())
    print(node.obj_dict)

for edge in graph.get_edges():
    print(edge.get_source())
    print(edge.get_destination())
    print(edge.to_string())
    print(edge.obj_dict)