PyQt与Matplotlib的实时画图案例

作者：渣渣辉2024.01.17 22:02浏览量：36

简介：使用PyQt和Matplotlib库，实现实时数据可视化。通过PyQt创建用户界面，使用Matplotlib进行数据绘图，展示如何实现实时数据流的可视化。

要实现实时画图，首先需要安装PyQt和Matplotlib库。你可以使用以下命令进行安装：

  1. pip install PyQt5 matplotlib

下面是一个简单的实时画图案例，我们将使用PyQt创建窗口，并使用Matplotlib进行绘图。在这个例子中，我们将创建一个简单的正弦波图。

  1. import sys
  2. import numpy as np
  3. from PyQt5.QtWidgets import QApplication, QMainWindow, QPushButton, QVBoxLayout, QWidget
  4. from PyQt5.QtCore import QTimer, pyqtSignal
  5. import matplotlib.pyplot as plt
  6. from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
  7. import matplotlib.animation as animation

接下来，我们定义一个类RealTimePlotter，继承自QMainWindow，实现实时画图功能。在这个类中，我们将使用Matplotlib的动画功能，实时更新图形。

  1. class RealTimePlotter(QMainWindow):
  2. def __init__(self):
  3. super().__init__()
  4. self.initUI()
  5. self.x = []
  6. self.y = []
  7. self.fig, self.ax = plt.subplots()
  8. self.line, = self.ax.plot([], [])
  9. self.timer = QTimer()
  10. self.timer.timeout.connect(self.update_plot)
  11. self.timer.start(1000)  # 1000ms更新一次图形

在__init__方法中，我们初始化用户界面，并创建了一个QTimer对象，用于定时更新图形。我们还创建了一个空的线对象self.line，用于在后续的动画更新中绘制图形。
接下来，我们定义了一个方法update_plot，用于更新图形。在这个方法中，我们将使用Matplotlib的动画更新函数FuncAnimation，实时更新图形。每次更新时，我们将添加一个新的点到数据集中，并使用这些数据绘制新的图形。

  1. def update_plot(self):
  2. x = np.arange(len(self.x))  # 计算x轴位置
  3. y = np.array(self.y)  # 获取当前y轴数据
  4. self.line.set_data(x, y)  # 更新线对象的数据
  5. self.ax.relim()  # 重新设置轴的范围，以适应新的数据点
  6. self.ax.autoscale_view()  # 自动缩放视图以适应新的数据点范围
  7. self.canvas.draw()  # 重新绘制图形

最后，我们定义一个方法add_data，用于添加新的数据点。我们可以将这个方法连接到按钮的点击事件上，以便在用户点击按钮时添加新的数据点。

  1. def add_data(self, value):
  2. self.x.append(len(self.x))  # 添加新的x轴位置
  3. self.y.append(value)  # 添加新的y轴数据点

在主程序中，我们创建一个RealTimePlotter对象，并显示它。我们还将创建一个按钮，并将其点击事件连接到add_data方法上。这样，每当用户点击按钮时，就会添加一个新的数据点到图形中。

最热文章