PyQt5中嵌入NetworkX关系图及matplotlib绘图自适应Pyqt

PyQt5中嵌入Matplotlib步骤如下:

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

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

第二步,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()

Matplotlib绘图自适应Pyqt5

QGraphicsScene无法放入布局中,导致绘图无法弹性布局
1、创建一个widget,在widget中加上QVBoxLayout
2、创建你的canvas,注意其父类是widget,不是QVBoxLayout
3、在QVBoxLayout中调用addWidget把canvas加进去

self.graphicsView = QtWidgets.QGraphicsView(self.tab_2)
self.qvbl = QtWidgets.QVBoxLayout(self.graphicsView)
# 判断布局中是否有控件
if self.qvbl.count() == 0:
    self.qvbl.addWidget(self.dr)
else:
    # 先清空布局中的所有控件,再添加新的控件
    for i in range(self.qvbl.count()):
        self.qvbl.itemAt(i).widget().deleteLater()
    self.qvbl.addWidget(self.dr)

QGrphicsView, QGraphicsScene 和 QGraphicsItem 的区别

1.Scene是一个场景,View是一个视野。如果视图大于场景时,场景在中间部分显示[如果不设置视图的Alignment属性];如果视图小于场景,则只能看到场景的一部分,但是会自动提供滚动条在整个场景中移动
2.QGraphicsView默认带有内边距和边界setStyleSheet("padding: 0px; border: 0px;")
3.QGrphicsView类实际上是为QGraphicsScene类的内容提空了一个控件,它在一个可滑动视图空间内可视化QGraphicsScene的内容。

QGraphicsScene类提供了一个平面来管理大量的2D的图像项目,像点,直线,多边形,圆等。它是QGraphicsItem类的容器。

打个比方来说,QGrphicsView类就像是一片空旷的场地,QGraphicsScene类是上面搭建的一个舞台,QGraphicsItem类是舞台上的各种东西

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

你可能感兴趣的:(PyQt5中嵌入NetworkX关系图及matplotlib绘图自适应Pyqt)