我有一个将PlotDataItem添加到特定绘图小部件的函数,但是,如果我尝试在绘图小部件上使用removeItem函数,则它实际上没有任何作用。我正在寻求有关如何使删除项在此特定情况下起作用的帮助?您可能会为优化,可读性等推荐的任何其他技巧也将不胜感激,因为我对PyQt甚至Python本身还是相当陌生的。谢谢!
此函数包括removeItem()函数。
def updateGraph(self):
"""Clears and updates graph to match the toggled checkboxes.
"""
# self.graphWidget.clear()
for checkboxNumber, checkbox in enumerate(
self.scenarioWidget.findChildren(QtWidgets.QCheckBox)
):
if checkbox.isChecked():
peak = self._model.get_peak(checkboxNumber)
duration = self._model.get_duration(checkboxNumber)
self.drawLine(
name=checkbox.objectName(),
peak=peak,
color=2 * checkboxNumber,
duration=duration,
)
else:
self.graphWidget.removeItem(pg.PlotDataItem(name=checkbox.objectName()))
# TODO: Allow for removal of individual pg.PlotDataItems via self.graphWidget.removeItem()
此功能是将PlotDataItems添加到绘图小部件的位置。
def drawLine(self, name, peak, color, duration=100.0):
"""Graphs sinusoidal wave off given 'peak' and 'duration' predictions to model epidemic spread.
Arguments:
name {string} -- Name of scenario/curve
peak {float} -- Predicted peak (%) of epidemic.
color {float} -- Color of line to graph.
Keyword Arguments:
duration {float} -- Predicted duration of epidemic (in days). (default: {100.0})
"""
X = np.arange(duration)
y = peak * np.sin((np.pi / duration) * X)
self.graphWidget.addItem(
pg.PlotDataItem(X, y, name=name, pen=pg.mkPen(width=3, color=color))
)
您正在使用pg.PlotDataItem(name=checkbox.objectName())
创建一个新对象,因此将找不到它,因为它是全新的。
未经测试,但应该可以工作:
for item in self.graphWidget.listDataItems():
if item.name() == checkbox.objectName():
self.graphWidget.removeItem(item)