如何使用计时器更新 axvline？

Question

我想使用 PyQt5 为绘图制作动画，以便垂直线随着时间的推移在我的图形上滚动。为了实现这一点，我开始修改this neat LearnPyQt tutorial中提供的代码示例。到目前为止，我能够使用以下方法在用户定义的横坐标处绘制一条垂直线：

self.canvas.axes.axvline(x=5)

如果我使用 for 循环，我会根据索引简单地更新 x，但在这种情况下，我不知道如何根据计时器 ID 更新 x。如何恢复此值并将其传递到我的 update_plot 函数中？

到目前为止，这是我的代码：

 import sys
 import random
 import matplotlib
 matplotlib.use('Qt5Agg')
 import pdb
 
 from PyQt5 import QtCore, QtWidgets
 
 from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
 from matplotlib.figure import Figure
 
 
 class MplCanvas(FigureCanvas):
 
     def __init__(self, parent=None, width=5, height=4, dpi=100):
         fig = Figure(figsize=(width, height), dpi=dpi)
         self.axes = fig.add_subplot(111)
         super(MplCanvas, self).__init__(fig)
 
 
 class MainWindow(QtWidgets.QMainWindow):
 
     def __init__(self, *args, **kwargs):
         super(MainWindow, self).__init__(*args, **kwargs)
 
         self.canvas = MplCanvas(self, width=5, height=4, dpi=100)
         self.setCentralWidget(self.canvas)
         self.timer_id = 0
         n_data = 50
         self.xdata = list(range(n_data))
         self.ydata = [random.randint(0, 10) for i in range(n_data)]
         self.update_plot()
 
         self.show()
 

         self.timer = QtCore.QTimer()
         self.timer.setInterval(100)
 
         self.timer.timeout.connect(self.update_plot)
         self.timer.start()
 
     def update_plot(self):
 
         self.canvas.axes.plot(self.xdata, self.ydata, 'r')
         self.canvas.axes.axvline(x=5)
         self.canvas.draw()
 
 
 app = QtWidgets.QApplication(sys.argv)
 w = MainWindow()
 app.exec_()

Answer 1

如果保留对 axvline（Line2D 对象）的引用，则可以调用 set_xdata 并在调用 update_plot 时递增 self.timer_id 以沿图形移动线。

class MainWindow(QtWidgets.QMainWindow):

    def __init__(self, *args, **kwargs):
        super(MainWindow, self).__init__(*args, **kwargs)

        self.canvas = MplCanvas(self, width=5, height=4, dpi=100)
        self.setCentralWidget(self.canvas)
        self.timer_id = 0
        n_data = 50
        self.xdata = list(range(n_data))
        self.ydata = [random.randint(0, 10) for i in range(n_data)]
        self.update_plot()

        self.show()


        self.timer = QtCore.QTimer()
        self.timer.setInterval(100)

        self.timer.timeout.connect(self.update_plot)
        self.timer.start()

    def update_plot(self):
        if not self.timer_id:
            self.canvas.axes.plot(self.xdata, self.ydata, 'r')
            self.vline = self.canvas.axes.axvline(x=self.timer_id)
        else:
            self.vline.set_xdata(self.timer_id)
        self.canvas.draw()
        self.timer_id += 1

如何使用计时器更新 axvline？

How can I update axvline using a timer?

python

timer

pyqt