使用networkx根据点和边绘制图

已知顶点数据（点ID、矢量值、点类型）和边/弧数据（索引对的数组，(from_id, to_id)），绘制其对应的图/树（有向图）。

# pip install networkx
import matplotlib.pyplot as plt
import networkx as nx
import numpy as np  

class draw_tree:
    def __init__(self):
        self.m_G = nx.Graph()  # 创建一个空的无向图
        self.m_nodes = [] # 节点
        self.m_edges = [] # 存放所有的边，构成无向图

    def set_nodes(self, nodes):
        self.m_nodes = nodes
        self.m_G.add_nodes_from(self.m_nodes)  # 从列表中加点

    def set_edges(self, edges):
        self.m_edges = edges
        self.m_G.add_edges_from(self.m_edges) # 将所有边加入网络

    def show(self, labels=None):
        nx.draw_networkx(self.m_G, labels=labels)
        plt.show()

"""
from my_draw_tree import draw_tree

num = 30
nodes = list(range(num))  # [0,1,2,3,4,5]

edges = []
for idx in range(num - 1):
    edges.append((idx, idx + 1))
edges.append((num - 1, 0))

dt = draw_tree()
dt.set_nodes(nodes)
dt.set_edges(edges)
dt.show()
"""

如果想要美化上述图，可以进一步探索nx.draw_networkx函数的其它参数，从而达到改变线条颜色、改变线条宽度、添加标签等效果。

from my_draw_tree import draw_tree

labels = {0:"A", 1:"B", 2:"C", 3:"D", 4:"E"}
dt.show(labels=l