在PyQT5中使用matplotlib复选按钮

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QVBoxLayout, QCheckBox, QWidget
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("PyQT5 with Matplotlib")
        self.setGeometry(100, 100, 800, 600)
        self.setupUI()

    def setupUI(self):
        self.figure = Figure()
        self.canvas = FigureCanvas(self.figure)
        self.ax = self.figure.add_subplot(111)

        self.checkbox = QCheckBox("显示/隐藏曲线", self)
        self.checkbox.stateChanged.connect(self.togglePlot)

        layout = QVBoxLayout()
        layout.addWidget(self.canvas)
        layout.addWidget(self.checkbox)

        widget = QWidget()
        widget.setLayout(layout)
        self.setCentralWidget(widget)

    def togglePlot(self, state):
        if state == 2:  # 选中状态
            self.ax.plot([1, 2, 3, 4], [1, 4, 9, 16])
        else:
            self.ax.clear()
        self.canvas.draw()

if __name__ == "__main__":
    app = QApplication(sys.argv)
    mainWindow = MainWindow()
    mainWindow.show()
    sys.exit(app.exec_())