在Matplotlib圖上獨立添加/刪除圖

Question

我想生成一個散點圖（最多500萬點），並在此之上添加不同的統計信息（例如Q1，中位數，Q3）。 這樣做的想法是添加/刪除這些統計信息，而無需重新繪制散點圖，從而加快流程。 到目前為止，我可以在圖上獨立添加圖，但是不能刪除特定圖。 取消選中該復選框時，出現以下錯誤：

AttributeError: 'Graphics' object has no attribute 'vline1'

我知道在創建圖時，我需要存儲/返回圖，以便以后在要刪除它時調用它，但我不知道該怎么做。

這是我當前的代碼：

import sys
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.pyplot import Figure

class Mainwindow(QMainWindow):
    def __init__(self, parent=None):
        super(Mainwindow, self).__init__(parent)

        centralWidget = QWidget()
        self.setCentralWidget(centralWidget)
        self.fig = Figure()
        self.axes = self.fig.add_subplot(111)
        self.canvas = FigureCanvas(self.fig)
        self.gridLayout = QGridLayout(centralWidget)
        self.gridLayout.addWidget(self.canvas)   
        self.btn_plot = QCheckBox("Plot")
        self.btn_line = QCheckBox("Line")
        self.gridLayout.addWidget(self.btn_plot, 1,0,1,1)
        self.gridLayout.addWidget(self.btn_line, 2,0,1,1)
        self.btn_plot.clicked.connect(self.btnPlot)
        self.btn_line.clicked.connect(self.btnLine)

    def btnPlot(self):
        self.checked = self.btn_plot.isChecked()
        self.Graphics = Graphics('plot', self.checked, self.axes)

    def btnLine(self):
        self.checked = self.btn_line.isChecked()
        self.Graphics = Graphics('line', self.checked, self.axes)

class Graphics:
    def __init__(self, typeGraph, checked, axes):
        self.typeGraph = typeGraph
        self.checked = checked
        self.axes = axes
        if self.typeGraph == 'plot': self.drawPlot()
        if self.typeGraph == 'line': self.drawLine()

    def drawPlot(self):
        if self.checked == True:
            self.plot = self.axes.plot([10,20,30], [5,10,2], 'o')
        else:
            self.plot.remove()
        self.axes.figure.canvas.draw()

    def drawLine(self):
        if self.checked == True:
            self.vline1 = self.axes.axvline(x=15, linestyle="dashed", color="#595959")
            self.vline2 = self.axes.axvline(x=25, linestyle="dashed", color="#595959")
        else:
            self.vline1.remove()
            self.vline2.remove()
        self.axes.figure.canvas.draw()

if __name__ == "__main__":
    app = QtWidgets.QApplication(sys.argv)
    prog = Mainwindow()   
    prog.show()
    sys.exit(app.exec_())

Answer 1

問題是因為當您單擊它時，它將始終創建不具有先前值的新Graphics （在btnPlot / btnLine ） plot, vline1, vline2 。 您只需創建一次Graphics僅運行drawPlot(checked) ， drawLine(checked)即可添加或刪除項目。

import sys
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.pyplot import Figure

class Mainwindow(QMainWindow):
    def __init__(self, parent=None):
        super(Mainwindow, self).__init__(parent)

        centralWidget = QWidget()
        self.setCentralWidget(centralWidget)
        self.fig = Figure()
        self.axes = self.fig.add_subplot(111)
        self.canvas = FigureCanvas(self.fig)
        self.gridLayout = QGridLayout(centralWidget)
        self.gridLayout.addWidget(self.canvas)   
        self.btn_plot = QCheckBox("Plot")
        self.btn_line = QCheckBox("Line")
        self.gridLayout.addWidget(self.btn_plot, 1,0,1,1)
        self.gridLayout.addWidget(self.btn_line, 2,0,1,1)
        self.btn_plot.clicked.connect(self.btnPlot)
        self.btn_line.clicked.connect(self.btnLine)

        # create only once
        self.Graphics = Graphics(self.axes)

    def btnPlot(self):
        # add or remove 
        self.Graphics.drawPlot(self.btn_plot.isChecked())

    def btnLine(self):
        # add or remove 
        self.Graphics.drawLine(self.btn_line.isChecked())

class Graphics:
    def __init__(self, axes):
        self.axes = axes
        # create at start with default values (but frankly, now I don't need it)
        self.plot = None
        self.vline1 = None
        self.vline2 = None

    def drawPlot(self, checked):
        if checked:
            self.plot = self.axes.plot([10,20,30], [5,10,2], 'o')
        else:
            for item in self.plot:
                item.remove()
        self.axes.figure.canvas.draw()

    def drawLine(self, checked):
        if checked:
            self.vline1 = self.axes.axvline(x=15, linestyle="dashed", color="#595959")
            self.vline2 = self.axes.axvline(x=25, linestyle="dashed", color="#595959")
        else:
            self.vline1.remove()
            self.vline2.remove()
        self.axes.figure.canvas.draw()

if __name__ == "__main__":
    app = QtWidgets.QApplication([])
    prog = Mainwindow()   
    prog.show()
    sys.exit(app.exec())