python - PyQt と Matplotlib: プロットのコピーを作成する

Question

私はmatplotlibが初めてで、かなり長い間機能の実装に苦労しています。基本的に、ウィジェットに埋め込まれたプロットがあり、(ボタンを押すと) このまったく同じプロットで新しいウィジェットを作成したいと考えています。これを行う簡単な方法はありますか？[すべてのアクション ハンドラーが既に実装されていると仮定して]

Answer 1

以下はembedding_in_qt4、matplotlib の例から切り取ったものです。

import sys
from PyQt4 import QtGui, QtCore

from numpy import arange, sin, pi
from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure


class MyMplCanvas(FigureCanvas):
    """Ultimately, this is a QWidget (as well as a FigureCanvasAgg, etc.)."""
    def __init__(self, parent=None, width=5, height=4, dpi=100):
        fig = Figure(figsize=(width, height), dpi=dpi)
        self.fig = fig
        self.axes = fig.add_subplot(111)
        # We want the axes cleared every time plot() is called
        self.axes.hold(False)

        self.compute_initial_figure()
        FigureCanvas.__init__(self, fig)
        self.setParent(parent)

        FigureCanvas.setSizePolicy(self,
                                   QtGui.QSizePolicy.Expanding,
                                   QtGui.QSizePolicy.Expanding)
        FigureCanvas.updateGeometry(self)

    def compute_initial_figure(self):
        pass


class MyStaticMplCanvas(MyMplCanvas):
    """Simple canvas with a sine plot."""
    def compute_initial_figure(self):
        t = arange(0.0, 3.0, 0.01)
        s = sin(2*pi*t)
        self.axes.plot(t, s)


class ApplicationWindow(QtGui.QWidget):
    def __init__(self):
        QtGui.QWidget.__init__(self)
        self.setAttribute(QtCore.Qt.WA_DeleteOnClose)
        self.main_widget = QtGui.QWidget(self)
        l = QtGui.QVBoxLayout(self)
        sc = MyStaticMplCanvas(self, width=5, height=4, dpi=100)
        self.sc = sc
        l.addWidget(sc)
        but = QtGui.QPushButton("make_new", self)
        but.clicked.connect(self.again)
        l.addWidget(but)

    def again(self):
        win = MyMplCanvas()
        win.fig = self.sc.fig
        FigureCanvas.__init__(win, win.fig)
        self.win = win
        win.show()

qApp = QtGui.QApplication(sys.argv)
aw = ApplicationWindow()
aw.show()
sys.exit(qApp.exec_())

キャンバスがその図オブジェクトへの参照を保持し、アプリケーションが新しいキャンバスへの参照を保持する方法に注意してください。