Обновление и переключение участков - PullRequest
Новая
       12

Обновление и переключение участков

1 голос
/ 28 июня 2019

Я пытаюсь обновить несколько графиков, используя встраивание matplotlibs в виджет QT. Прямо сейчас у меня может быть одно обновление графика в окне. Но когда я пытаюсь переключиться на другой график, нажав кнопку, программа зависает.

Это тестовый скрипт, который я использую, чтобы понять, как использовать инструмент программирования для интеграции в большую программу. Я изменил код из этого вопроса: Как встроить matplotlib в pyqt - для чайников

Я застрял на этой проблеме некоторое время. Я знаю, что мне не хватает чего-то очень простого. 


import random
import sys
from PyQt4 import QtGui, QtCore

from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.backends.backend_qt4agg import NavigationToolbar2QT as NavigationToolbar
from matplotlib.figure import Figure

class Window(QtGui.QDialog):
    def __init__(self, parent=None):
        super(Window, self).__init__(parent)

        # a figure instance to plot on
        self.figure1 = Figure()
        self.figure2 = Figure()
        self.current = "fig1"
        # this is the Canvas Widget that displays the `figure`
        # it takes the `figure` instance as a parameter to __init__
        self.canvas = FigureCanvas(self.figure1)
        self.ax1 = self.figure1.add_subplot(111)
        self.ax2 = self.figure2.add_subplot(111)
        self.line1, = self.ax1.plot([], [], 'r', lw=2)
        self.line2, = self.ax2.plot([], [], 'b', lw=2)
        # this is the Navigation widget
        # it takes the Canvas widget and a parent
        self.toolbar = NavigationToolbar(self.canvas, self)

        # Just some button connected to `plot` method
        self.button = QtGui.QPushButton('Plot')
        self.button.clicked.connect(self.plot)

        # set the layout
        layout = QtGui.QVBoxLayout()
        layout.addWidget(self.toolbar)
        layout.addWidget(self.canvas)
        layout.addWidget(self.button)
        self.setLayout(layout)
        self.update()


    def update(self):
        datax = [random.random() for i in range(10)]
        datay = [random.random() for i in range(10)]

        self.line1.set_xdata(datax)
        self.line1.set_ydata(datay)
        self.ax1.relim()
        self.ax1.autoscale_view()


        self.line2.set_xdata(datax)
        self.line2.set_ydata(datay)
        self.ax2.relim()
        self.ax2.autoscale_view()
        self.canvas.draw()
        QtCore.QTimer.singleShot(1, self.update)




    def plot(self):

        if self.current == "fig1":
            self.canvas = FigureCanvas(self.figure2)
            self.current = "fig2"
        elif self.current == "fig2":
            self.canvas = FigureCanvas(self.figure1)
            self.current = "fig1"


if __name__ == '__main__':
    app = QtGui.QApplication(sys.argv)

    main = Window()
    main.show()

    sys.exit(app.exec_())

Когда я нажимаю эту кнопку, она должна начать строить другой график. Я не получаю сообщений об ошибках.

1 Ответ

0 голосов
/ 28 июня 2019

Ваша логика изменения холста: 

def plot(self):
    if self.current == "fig1":
        self.canvas = FigureCanvas(self.figure2)
        self.current = "fig2"
    elif self.current == "fig2":
        self.canvas = FigureCanvas(self.figure1)
        self.current = "fig1"

Это похоже на следующую логику: 

class Foo:
    def __init__(self):
        self.m_a = 0

    def setA(self, a):
        self.m_a = a

    def printA(self):
        print(self.m_a)


class Bar:
    def __init__(self):
        self.a = 5
        self.foo = Foo()
        self.foo.setA(self.a)

    def change(self):
        self.a = 6


if __name__ == "__main__":
    bar = Bar()
    bar.foo.printA()
    bar.change()
    bar.foo.printA()

И если код выполняется, вы получаете следующее: 

5
5

Как видите, изменение переменной не означает, что другие классы уведомляются, потому что имя переменной является псевдонимом, важна память, в вашем случае отображаемый холст занимает место в памяти, отличное от холст, который вы создаете при нажатии кнопки, то же самое происходит с line1, line2, ax1 и ax2, которые связаны с исходным холстом.

С другой стороны, в графическом интерфейсе не рекомендуется удалять и создавать виджеты, вместо этого лучше скрыть виджет и показать другой, который занимает то же место, и для этого мы можем использовать QStackedWidget или QStackedLayout. 

import sys
import random
from PyQt4 import QtGui, QtCore

from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.backends.backend_qt4agg import (
    NavigationToolbar2QT as NavigationToolbar,
)
from matplotlib.figure import Figure


class Window(QtGui.QDialog):
    def __init__(self, parent=None):
        super(Window, self).__init__(parent)

        self.figure1 = Figure()
        self.canvas1 = FigureCanvas(self.figure1)
        self.ax1 = self.figure1.add_subplot(111)
        self.line1, = self.ax1.plot([], [], "r", lw=2)
        self.toolbar1 = NavigationToolbar(self.canvas1, self)

        self.figure2 = Figure()
        self.canvas2 = FigureCanvas(self.figure2)
        self.ax2 = self.figure2.add_subplot(111)
        self.line2, = self.ax2.plot([], [], "b", lw=2)
        self.toolbar2 = NavigationToolbar(self.canvas2, self)

        self.m_stacked_layout = QtGui.QStackedLayout()

        for canvas, toolbar in (
            (self.canvas1, self.toolbar1),
            (self.canvas2, self.toolbar2),
        ):
            widget = QtGui.QWidget()
            lay = QtGui.QVBoxLayout(widget)
            lay.addWidget(toolbar)
            lay.addWidget(canvas)
            self.m_stacked_layout.addWidget(widget)

        self.button = QtGui.QPushButton("Plot", clicked=self.onClicled)

        layout = QtGui.QVBoxLayout(self)
        layout.addLayout(self.m_stacked_layout)
        layout.addWidget(self.button)

        timer = QtCore.QTimer(self, timeout=self.update_plot, interval=1)
        timer.start()
        self.update_plot()

    @QtCore.pyqtSlot()
    def update_plot(self):
        datax = [random.random() for i in range(10)]
        datay = [random.random() for i in range(10)]

        self.line1.set_xdata(datax)
        self.line1.set_ydata(datay)
        self.ax1.relim()
        self.ax1.autoscale_view()

        self.line2.set_xdata(datax)
        self.line2.set_ydata(datay)
        self.ax2.relim()
        self.ax2.autoscale_view()
        self.canvas1.draw()
        self.canvas2.draw()

    @QtCore.pyqtSlot()
    def onClicled(self):
        ix = self.m_stacked_layout.currentIndex()
        self.m_stacked_layout.setCurrentIndex(0 if ix == 1 else 1)


if __name__ == "__main__":
    app = QtGui.QApplication(sys.argv)

    main = Window()
    main.show()

    sys.exit(app.exec_())
...