在PyQT5中使用matplotlib复选按钮

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

class MyApp(QMainWindow):
    def __init__(self):
        super().__init__()

        self.setWindowTitle("Matplotlib with PyQt5 Checkboxes")
        self.setGeometry(100, 100, 800, 600)

        # 创建主窗口的中心部件
        central_widget = QWidget(self)
        self.setCentralWidget(central_widget)

        # 创建布局
        layout = QVBoxLayout(central_widget)

        # 创建matplotlib图表
        self.figure = Figure()
        self.canvas = FigureCanvas(self.figure)
        layout.addWidget(self.canvas)

        # 添加复选按钮
        self.checkbox1 = QCheckBox("Series 1")
        self.checkbox2 = QCheckBox("Series 2")
        self.checkbox1.stateChanged.connect(self.update_plot)
        self.checkbox2.stateChanged.connect(self.update_plot)
        layout.addWidget(self.checkbox1)
        layout.addWidget(self.checkbox2)

        # 初始化图表
        self.ax = self.figure.add_subplot(111)
        self.data1 = [1, 3, 2, 4, 5]
        self.data2 = [5, 4, 3, 2, 1]
        self.update_plot()

    def update_plot(self):
        self.ax.clear()
        if self.checkbox1.isChecked():
            self.ax.plot(self.data1, label='Series 1')
        if self.checkbox2.isChecked():
            self.ax.plot(self.data2, label='Series 2')
        self.ax.legend()
        self.canvas.draw()

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