NetworkX 和 wxPython - 如何添加边权重？

根据您问题中提供的信息，我做出了以下假设：您要确保边缘长度基于边缘权重你想突出主节点现在让我们从第一个开始。1. 根据边权绘制边长您可以使用pygraphviz 和networkx 以边长根据您的选择来获得图形的布局。为此，首先您需要缩放边权重，以便节点之间的距离在图中可见（假定您的最小值为0.1和最大值为1，则边将不可见）。我将它们缩放了 10，但您可以使用适合您的任何缩放方法。new_edge_weights = [ int(x*10) for x in edge_weights]接下来，创建字典，键为“len”（pygraphviz 将使用此属性），值为边长lengths = {}for e,l in zip(edges, new_edge_weights): lengths[e] = dict(len=l)# {(1, 0): {'len': 10},# (2, 0): {'len': 5},# (3, 1): {'len': 5},# (4, 1): {'len': 1},# (5, 1): {'len': 1},# (6, 2): {'len': 1},# (7, 2): {'len': 1}}现在使用 graphviz_layout 获取节点的位置并绘制图形pos = graphviz_layout(G, prog='neato')nx.draw_networkx(G, pos=pos, node_size = node_sizes, node_color = node_color, edge_weights='len', labels = node_label, with_labels=True)这是您的图形最终的样子：2. 突出显示您的主节点（假设主节点是node 0: Printer。这可能有点棘手。因此，一种方法是在主节点本身上绘制另一个尺寸更小、颜色不同的节点。# Set the nodel label to empty for the new nodenode_label[len(node_label)+1] = "" # Set the size of the new node smaller than the master nodemaster_idx = 0node_sizes.append(node_sizes[master_idx]-300)# change the master node color to blacknode_color[master_idx] = "#000000"# set the color of the inner node (new smaller node)node_color.append("#00d992")# Set the position of the node same as the master nodepos[len(pos)+1] = pos[master_idx]# Add node to the graphG.add_node(len(node_label))# Draw the graphnx.draw_networkx(G, pos=pos, node_size = node_sizes, node_color = node_color, edge_weights='len', labels = node_label, with_labels=True)请注意，我已缩放节点大小以使边缘可见。您可以根据您的要求更改它并相应地调整边缘长度。Here is the full programimport wximport matplotlib.pyplot as pltimport networkx as nxfrom matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as FigCanvasfrom networkx.drawing.nx_agraph import graphviz_layoutclass NetworkFrame(wx.Frame): def __init__(self): wx.Frame.__init__(self, None, -1) self.SetSize(wx.Size(1280, 768)) self.panel = wx.Panel(self) self.fig = plt.figure() self.canvas = FigCanvas(self.panel, -1, self.fig) G = nx.Graph() nodes = [0,1,2,3,4,5,6,7] node_sizes = [650,400,400,250,250,250,250,250] node_sizes = [x+1500 for x in node_sizes] node_color = ["#00d992","#00d9c8","#00d9c8","#00b4d9","#00b4d9","#00b4d9","#00b4d9","#00b4d9"] edges = [(1,0),(2,0),(3,1),(4,1),(5,1),(6,2),(7,2)] node_label = {0: "Printer", 1: "Case", 2: "Electronics", 3: "Plastic 1", 4: "Plastic 2", 5: "Plastic 3", 6: "Metal 1", 7: "Metal 2"} edge_weights = [1, 0.5, 0.5, 0.1, 0.1, 0.1, 0.1, 0.1] # ????? new_edge_weights = [ int(x*10) for x in edge_weights] lengths = {} for e,l in zip(edges, new_edge_weights): lengths[e] = dict(len=l) G.add_nodes_from(nodes) G.add_edges_from(edges) nx.set_edge_attributes(G, lengths) pos = graphviz_layout(G, prog='neato') # Set the nodel label to empty for the new node node_label[len(node_label)+1] = "" # Set the size of the new node smaller than the master node master_idx = 0 node_sizes.append(node_sizes[master_idx]-300) # change the master node color to black node_color[master_idx] = "#000000" # set the color of the inner node (new smaller node) node_color.append("#00d992") # Set the position of the node same as the master node pos[len(pos)+1] = pos[master_idx] # Add node to the graph G.add_node(len(node_label)) # Draw the graph nx.draw_networkx(G, pos=pos, node_size = node_sizes, node_color = node_color, edge_weights='len', labels = node_label, with_labels=True) self.vbox = wx.BoxSizer(wx.VERTICAL) self.vbox.Add(self.canvas, 1, wx.LEFT | wx.TOP | wx.GROW) self.panel.SetSizer(self.vbox)if __name__ == '__main__': app = wx.App() app.frame = NetworkFrame() app.frame.Show() app.MainLoop()

NetworkX 和 wxPython - 如何添加边权重？

2回答