Question

我有一些嘈杂的数据，试图与高斯拟合。问题是我必须手动进行。这样，我的意思是我必须在曲线上移动该点（请参见下图）。当我移动点时，我必须更新曲线，以便其自身的曲线可以移动。

例如，在此曲线上，如果我移动较高的点，它将改变高斯的mu；如果我移动中间的点，它将更新sigma参数。在此示例中，我在嵌入在QMainWindow中的matplotlib的FigureCanvas中绘制了两条曲线。

我已经发现并没有办法在嵌入PyQt小部件中的matplotlib图形中执行此操作。因此，我进行了更改，并尝试将PyQtGraph与ROI工具一起使用，但是效果不是很好。

您知道我如何实现这一目标吗？有一个简单的python库可以做到这一点吗？谢谢

编辑：这是我用来生成图像的代码：

from PySide2 import QtCore, QtGui, QtWidgets

from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure
import numpy as np


class PainterCanvas(FigureCanvas):
    def __init__(self, parent=None, width=5, height=4, dpi=100):
        fig = Figure(figsize=(width, height), dpi=dpi)
        FigureCanvas.__init__(self, fig)
        self.setParent(parent)
        self._instructions = []
        self.axes = self.figure.add_subplot(111)

    def paintEvent(self, event):
        super().paintEvent(event)
        painter = QtGui.QPainter(self)
        painter.setRenderHint(QtGui.QPainter.Antialiasing, True)
        width, height = self.get_width_height()
        for x, y, rx, ry, br_color in self._instructions:
            x_pixel, y_pixel_m = self.axes.transData.transform((x, y))
            # In matplotlib, 0,0 is the lower left corner,
            # whereas it's usually the upper right
            # for most image software, so we'll flip the y-coor
            y_pixel = height - y_pixel_m
            painter.setBrush(QtGui.QColor(br_color))
            painter.drawEllipse( QtCore.QPoint(x_pixel, y_pixel), rx, ry)

    def create_oval(self, x, y, radius_x=2, radius_y=2, brush_color="red"):
        self._instructions.append([x, y, radius_x, radius_y, brush_color])
        self.update()


class MyPaintWidget(QtWidgets.QWidget):
    def __init__(self):
        super().__init__()

        self.canvas = PainterCanvas()
        self.canvas.mpl_connect("button_press_event", self._on_left_click)
        x = np.arange(0, 10, 0.1)
        rand = [np.random.uniform(-0.1, 0.2) for _ in x]
        y0 = np.exp(- (x - 5) ** 2 / 2) + rand
        y1 = np.exp(- (x - 3) ** 2 / 0.5)
        self.canvas.axes.plot(x, y0)
        self.canvas.axes.plot(x, y1)

        layout_canvas = QtWidgets.QVBoxLayout(self)
        layout_canvas.addWidget(self.canvas)

        self.canvasMenu = QtWidgets.QMenu(self)
        self.canvasMenu.addAction("test")

        self.canvas.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
        self.canvas.customContextMenuRequested.connect(self._on_left_click)

    def _on_left_click(self, event):
        self.canvas.create_oval(event.xdata, event.ydata, brush_color="green")


if __name__ == "__main__":
    import sys

    app = QtWidgets.QApplication(sys.argv)

    w = MyPaintWidget()
    w.show()
    sys.exit(app.exec_())

要添加点，我已经通过单击曲线来添加它们。我知道代码无法满足我的要求，而只是产生一张图像来解释我的想法。

Answer 1

如mkrieger1所建议，我将简要描述我的解决方案：

首先，我使用PyQtGraph绘制数据以比使用Matplotlib时更有效地绘制更新。
为控制曲线，我使用了QSliders（修改为支持double）来控制每个曲线的参数。当我移动一个滑块时，它将发出一个事件，并且在处理该事件的函数中，我使用函数setData更新了pyqtgraph plotwidget。

我的灵感来自做我的双重光标：Use float for QSlider 并且我使用了pyqtgraph文档：http://www.pyqtgraph.org/documentation/graphicsItems/plotdataitem.html

有太多代码需要放置，但解决方案在我的GitHub上的spectrofit项目中：https://github.com/fulmen27/SpectroFit（用于处理和研究恒星光谱的UI）。解决方案在spectrofit.tools.Interactive_Fit.py文件中。

希望这会有所帮助！

通过在曲线上移动一点来更新曲线

1 个答案: