美文网首页编程-Python
PyQt5中嵌入NetworkX关系图

PyQt5中嵌入NetworkX关系图

作者: 还是那个没头脑 | 来源:发表于2019-08-13 15:49 被阅读2次

    PyQt5中嵌入Matplotlib步骤如下:

    第一步,通过matplotlib.backends.backend_qt5agg类来连接PyQt5:

    from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
    from matplotlib.figure import Figure
    

    第二步,networdX画具体图形代码实现部分,例:

    class Figure_Canvas(FigureCanvas):
    # 通过继承FigureCanvas类,使得该类既是一个PyQt5的Qwidget,又是一个matplotlib的FigureCanvas,这是连接pyqt5与matplotlib的关键
        def __init__(self):
            self.fig = plt.figure()# 可选参数,facecolor为背景颜色
            FigureCanvas.__init__(self, self.fig)  # 初始化激活widget中的plt部分
    
        def draw_graph(self):
            node_color = ['r']
            data = {'张三':
                        {'校友': ['李一', '李二', '李三', '李四'],
                         '同学': ['王一', '王二', '王三', '王四', '王五'],
                         '同事': ['马一', '马二', '马三', '马四', '马五', '马六']
                         }
                    }
    
            self.G = nx.Graph()
            for k, v in data.items():
                self.G.add_node(k, id='0001')
                for m, n in v.items():
                    if m == '校友':
                        node_color.extend(['b'] * len(n))
                        for i in n:
                            self.G.add_node(i, id='0002')
                            self.G.add_edge(k, i, weight=0.3, relation='校友')
                    elif m == '同学':
                        node_color.extend(['b'] * len(n))
                        for i in n:
                            self.G.add_node(i, id='0003')
                            self.G.add_edge(k, i, weight=0.5, relation='同学')
                    else:
                        node_color.extend(['g'] * len(n))
                        for i in n:
                            self.G.add_node(i, id='0003')
                            self.G.add_edge(k, i, weight=0.9, relation='同事')
            pos = nx.spring_layout(self.G)
    
            relations = {}
            for po in pos:
                print(po, pos[po])
                xy = [[pos[po][0] - 0.05, pos[po][0] + 0.05],
                      [pos[po][1] - 0.05, pos[po][1] + 0.05]]
                relations[po] = xy
    
            labels = {}
            for edge in self.G.edges(data=True):
                labels[(edge[0], edge[1])] = edge[-1]['relation']
            #如果不使用canvas.mpl_connect的话将不能激活event
            self.fig.canvas.mpl_connect('button_press_event', lambda event: self.on_press(event, relations))
    
            nx.draw(self.G, pos,with_labels=True, node_color=node_color, edge_color='b', node_size=1000,
                    rotate=True)  
    
            nx.draw_networkx_edge_labels(self.G, pos, edge_labels=labels)
    

    第三步,GUI上通过控件呈现matplotlib画出来的图形——通过QtWidgets.QGraphicsView控件来实现:(代码部分仅呈现QtWidgets.QGraphicsView的实现步骤)

    # 第一步,创建一个QGraphicsView,注意同样以gridLayoutWidget为参数
    self.graphicview = QtWidgets.QGraphicsView(self.gridLayoutWidget)
    # 第二步,设置graphicview的区域
    self.graphicsView.setGeometry(QtCore.QRect(40, 10, 481, 431))
    # 第三步,实例化一个FigureCanvas
    dr = Figure_Canvas()
    dr.draw_graph() # 画图
    # 第四步,创建一个QGraphicsScene,因为加载的图形(FigureCanvas)不能直接放到graphicview控件中,必须先放到graphicScene,然后再把graphicscene放到graphicview中
    graphicscene = QtWidgets.QGraphicsScene()
    # 第五步,把图形放到QGraphicsScene中,注意:图形是作为一个QWidget放到QGraphicsScene中的
    graphicscene.addWidget(dr)
    # 第六步,把QGraphicsScene放入QGraphicsView
    self.graphicview.setScene(graphicscene)
    # 第七步,调用show方法呈现图形!
    self.graphicview.show()
    

    参考:
    https://www.cnblogs.com/laoniubile/p/5904817.html
    https://blog.csdn.net/qq_31809257/article/details/89292824

    相关文章

      网友评论

        本文标题:PyQt5中嵌入NetworkX关系图

        本文链接:https://www.haomeiwen.com/subject/secbjctx.html