如何更新 pyqtgraph 中的绘图？

我正在尝试使用 PyQt5 和 pyqtgraph 拥有一个用户界面。我制作了两个复选框，每当我选择它们时，我想绘制代码中可用的两个数据集之一，每当我取消选择按钮时，我希望它清除相应的曲线。有两个带有文本的复选框A1 and A2他们每个人都绘制一组数据。

我有两个问题：

1-如果我选择A1它绘制与相关的数据A1只要我不选择A2，通过取消选择A1我可以清除相关数据A1。 但是，如果我检查A1框，然后我检查A2框，然后取消选择A1不清除相关情节。在这种情况下，如果我选择绘制随机数据，而不是确定性曲线，例如sin，我看到通过选择任一按钮都会添加新数据，但无法删除它。

2- 真实的应用程序有 96 个按钮，每个按钮都应与一个数据集相关联。我认为我编写代码的方式效率很低，因为我需要为一个按钮和数据集复制相同的代码 96 次。有没有办法将我下面介绍的玩具代码概括为任意数量的复选框？或者，为每个按钮使用/复制几乎相同的代码可能是执行此操作的常用且正确的方法？

代码是：

from PyQt5 import QtWidgets, uic, QtGui
import matplotlib.pyplot as plt
from matplotlib.widgets import SpanSelector
import numpy as np
import sys
import string
import pyqtgraph as pg
from pyqtgraph.Qt import QtGui, QtCore

app = QtWidgets.QApplication(sys.argv)

x = np.linspace(0, 3.14, 100)
y1 = np.sin(x)#Data number 1 associated to checkbox A1
y2 = np.cos(x)#Data number 2 associated to checkbox A2

#This function is called whenever the state of checkboxes changes
def todo():
    if cbx1.isChecked():
        global curve1
        curve1 = plot.plot(x, y1, pen = 'r')
    else:
        try:
            plot.removeItem(curve1)
        except NameError:
            pass
    if cbx2.isChecked():
        global curve2
        curve2 = plot.plot(x, y2, pen = 'y')
    else:
        try:
            plot.removeItem(curve2)
        except NameError:
            pass  
#A widget to hold all of my future widgets
widget_holder = QtGui.QWidget()

#Checkboxes named A1 and A2
cbx1 = QtWidgets.QCheckBox()
cbx1.setText('A1')
cbx1.stateChanged.connect(todo)

cbx2 = QtWidgets.QCheckBox()
cbx2.setText('A2')
cbx2.stateChanged.connect(todo)

#Making a pyqtgraph plot widget
plot = pg.PlotWidget()

#Setting the layout
layout = QtGui.QGridLayout()
widget_holder.setLayout(layout)

#Adding the widgets to the layout
layout.addWidget(cbx1, 0,0)
layout.addWidget(cbx2, 0, 1)
layout.addWidget(plot, 1,0, 3,1)

widget_holder.adjustSize()
widget_holder.show()

sys.exit(app.exec_())


    

下面是我制作的一个示例，效果很好。 可以重复使用做更多的绘图，无需增加代码，只需更改self.num并使用函数添加相应的数据add_data(x,y,ind), where x and y是数据的值，ind是盒子的索引（来自0 to n-1).

import sys
import numpy as np
import pyqtgraph as pg
from pyqtgraph.Qt import QtCore, QtGui

class MyApp(QtGui.QWidget):
    def __init__(self):
        QtGui.QWidget.__init__(self)
        self.central_layout = QtGui.QVBoxLayout()
        self.plot_boxes_layout = QtGui.QHBoxLayout()
        self.boxes_layout = QtGui.QVBoxLayout()
        self.setLayout(self.central_layout)
        
        # Lets create some widgets inside
        self.label = QtGui.QLabel('Plots and Checkbox bellow:')
        
        # Here is the plot widget from pyqtgraph
        self.plot_widget = pg.PlotWidget()
        
        # Now the Check Boxes (lets make 3 of them)
        self.num = 6
        self.check_boxes = [QtGui.QCheckBox(f"Box {i+1}") for i in range(self.num)]
        
        # Here will be the data of the plot
        self.plot_data = [None for _ in range(self.num)]
        
        # Now we build the entire GUI
        self.central_layout.addWidget(self.label)
        self.central_layout.addLayout(self.plot_boxes_layout)
        self.plot_boxes_layout.addWidget(self.plot_widget)
        self.plot_boxes_layout.addLayout(self.boxes_layout)
        for i in range(self.num):
            self.boxes_layout.addWidget(self.check_boxes[i])
            # This will conect each box to the same action
            self.check_boxes[i].stateChanged.connect(self.box_changed)
            
        # For optimization let's create a list with the states of the boxes
        self.state = [False for _ in range(self.num)]
        
        # Make a list to save the data of each box
        self.box_data = [[[0], [0]] for _ in range(self.num)] 
        x = np.linspace(0, 3.14, 100)
        self.add_data(x, np.sin(x), 0)
        self.add_data(x, np.cos(x), 1)
        self.add_data(x, np.sin(x)+np.cos(x), 2)
        self.add_data(x, np.sin(x)**2, 3)
        self.add_data(x, np.cos(x)**2, 4)
        self.add_data(x, x*0.2, 5)
        

    def add_data(self, x, y, ind):
        self.box_data[ind] = [x, y]
        if self.plot_data[ind] is not None:
            self.plot_data[ind].setData(x, y)

    def box_changed(self):
        for i in range(self.num):
            if self.check_boxes[i].isChecked() != self.state[i]:
                self.state[i] = self.check_boxes[i].isChecked()
                if self.state[i]:
                    if self.plot_data[i] is not None:
                        self.plot_widget.addItem(self.plot_data[i])
                    else:
                        self.plot_data[i] = self.plot_widget.plot(*self.box_data[i])
                else:
                    self.plot_widget.removeItem(self.plot_data[i])
                break
        
if __name__ == "__main__":
    app = QtGui.QApplication(sys.argv)
    window = MyApp()
    window.show()
    sys.exit(app.exec_())

请注意，内部德PlotWidget我使用添加情节plot()方法，它返回一个PlotDataItem保存在调用之前创建的列表中self.plot_data。 有了这个，您可以轻松地将其从Plot Widget并再次添加。此外，如果您的目标是更复杂的程序，例如，您可以在运行时更改每个框的数据，那么如果您使用setData()方法上的PlotDataItem

正如我在一开始所说的，这应该适用于很多复选框，因为当复选框被选中/取消选中时调用的函数，首先将每个框的实际状态与前一个框进行比较（存储在self.state）并且只对与该特定框相对应的图进行更改。这样，您就可以避免为每个复选框执行一个函数，并避免每次选中/取消选中某个框时重新绘制所有 de 框（例如用户8408080 https://stackoverflow.com/users/8408080/user8408080做过）。我并不是说这不好，但如果增加复选框的数量和/或数据的复杂性，重新绘制所有数据的工作量将急剧增加。

唯一的问题是，当窗口太小而无法支持大量复选框（例如 96 个）时，您将不得不在另一个小部件而不是布局中组织复选框。

Now some screenshots of the code from above:

然后改变值self.num to 6并向其中添加一些随机数据：

self.add_data(x, np.sin(x)**2, 3)
self.add_data(x, np.cos(x)**2, 4)
self.add_data(x, x*0.2, 5)