程序在边缘绘制权重不能正常工作+ networkx

时间:2011-05-01 16:57:12

标签: python networkx

我有这个程序在wx帧中使用matplotlib在节点上绘制带标签的边缘。

我已经使用网站上的示例和其他人提出的查询进行了组合。

但它没有正常工作,因为节点和边缘确实被绘制但权重没有。

有人可以帮我找到原因......

import wxversion
wxversion.ensureMinimal('2.8')

import matplotlib

matplotlib.use('WXAgg')
from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as FigureCanvas

from matplotlib.backends.backend_wx import NavigationToolbar2Wx

from matplotlib.figure import Figure

import wx

import networkx as nx

class CanvasFrame(wx.Frame):

  def __init__(self):
    wx.Frame.__init__(self,None,-1,
                     'CanvasFrame',size=(550,350))

    self.SetBackgroundColour(wx.NamedColor("WHITE"))

    self.figure = Figure()
    self.axes = self.figure.add_subplot(111)

    self.canvas = FigureCanvas(self, -1, self.figure)

    self.sizer = wx.BoxSizer(wx.VERTICAL)
    self.sizer.Add(self.canvas, 1, wx.LEFT | wx.TOP | wx.GROW)
    self.SetSizer(self.sizer)
    self.Fit()
    G = nx.Graph()
    G.add_edge(1,3,weight = 5)
    G.add_edge(1,2,weight = 4)   

    pos = nx.spring_layout(G)    
    nx.draw_networkx(G, pos, ax=self.axes)
    edge_labels=dict([((u,v,),d['weight'])
         for u,v,d in G.edges(data=True)])
    nx.draw_networkx_edge_labels(G,pos,edge_labels=edge_labels)
class App(wx.App):

  def OnInit(self):
    'Create the main window and insert the custom frame'
    frame = CanvasFrame()
    frame.Show(True)

    return True

app = App(0)
app.MainLoop()

1 个答案:

答案 0 :(得分:2)

对此答案的网络社区Mr Aric Hagberg的信用。

然而,自从我开始了解它以来,我认为我应该在这里为更多用户回答。

上述代码中唯一的问题是

nx.draw_networkx_edge_labels(G,pos,edge_labels=edge_labels,ax= self.axes)

现在它使用networkx在wx内部给出加权边。