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

Question

我试图找出如何为我的代码示例添加边缘权重，但我找不到任何与 wxPython 结合的好的文档。

我想要边缘，可以说节点的距离与我的列表“edge_weight”中的一样。标签和其他一切都按预期工作。

作为一个小的补充，是否可以添加类似阴影效果或任何类似于主节点的东西来突出它？

我的代码如下所示：

import wx
import matplotlib.pyplot as plt
import networkx as nx
from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as FigCanvas

class 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 = [6500,4000,4000,2500,2500,2500,2500,2500]
        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]  # ?????             
        
        G.add_nodes_from(nodes)
        G.add_edges_from(edges)
        
        nx.draw(G, node_size = node_sizes, node_color = node_color, labels = node_label, with_labels=True)  
        
        plt.axis('off')
        
        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()

Answer 1

使用 add_weighted_edges_from 代替 add_edges_from

import wx
import matplotlib.pyplot as plt
import networkx as nx
from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as FigCanvas


class 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 = [6500, 4000, 4000, 2500, 2500, 2500, 2500, 2500]
    node_color = ["#00d992", "#00d9c8", "#00d9c8", "#00b4d9", "#00b4d9", "#00b4d9", "#00b4d9", "#00b4d9"]
    edges = [(1, 0, 1), (2, 0, 0.5), (3, 1, 0.1), (4, 1, 0.1), (5, 1, 0.1), (6, 2, 0.1), (7, 2, 0.1)]
    node_label = {0: "Printer",
                  1: "Case",
                  2: "Electronics",
                  3: "Plastic 1",
                  4: "Plastic 2",
                  5: "Plastic 3",
                  6: "Metal 1",
                  7: "Metal 2"}

    G.add_nodes_from(nodes)
    G.add_weighted_edges_from(edges)

    nx.draw(G, node_size=node_sizes, node_color=node_color, labels=node_label, with_labels=True)

    plt.axis('off')

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

Answer 2

我根据您问题中提供的信息做出以下假设：

您想确保边长基于边权重
你要高亮master节点

现在让我们从第一个开始。

1.根据边权重绘制边长 您可以将 pygraphviz 与 networkx 一起使用，以根据您的选择选择边长的方式来获取图形的布局。

为此，首先您需要缩放边权重，以便节点之间的距离在图中可见（假定您的最小值为 0.1，最大值为1，边缘将不可见）。我已经将它们缩放了 10，但是您可以使用适合您的任何缩放方法。

new_edge_weights = [ int(x*10) for x in edge_weights]

接下来，创建字典，键为 'len'（pygraphviz 将使用此属性，参见 here），值为边长

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

.
请注意，我已缩放节点大小以使边缘可见。您可以根据您的要求更改它并相应地调整边长。

Here is the full program
import wx
import matplotlib.pyplot as plt
import networkx as nx
from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as FigCanvas
from networkx.drawing.nx_agraph import graphviz_layout


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

参考资料

graphviz_layout
pygraphviz
Set edge lengths according to weights in graphviz_layout

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

NetworkX and wxPython - How to add edge weight?

python

wxpython

graph

networkx

python-3.x