显示滑动窗口中数据的平均值,随着窗口的滑动,数据更新,平均值随之更新
代码实现
库导入
import sys
import random
import numpy as np
import matplotlib
from PyQt5.QtWidgets import QApplication, QMainWindow, QVBoxLayout, QWidget
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure
from matplotlib.animation import FuncAnimation
matplotlib.rcParams['font.sans-serif'] = ['SimHei']
matplotlib.rcParams['axes.unicode_minus'] = False
MyMainWindow类(功能主体部分)
初始化函数
def __init__(self):
super().__init__()
# 设置主窗口的标题和大小
self.setWindowTitle("实时随机数绘图")
self.setGeometry(100, 100, 800, 600)
# 初始化UI
self.setup_ui()
setup_ui 函数
功能
设置主窗口的用户界面,包括绘图区域、绘图线、动画对象等的初始化。
具体包括:
-
创建一个 QWidget 对象作为主窗口的中心部件。将该中心部件设置为主窗口的中心部件。
-
创建一个 QWidget 对象作为主窗口的中心部件。将该中心部件设置为主窗口的中心部件。
-
创建一个 QVBoxLayout 对象作为中心部件的布局管理器。
-
创建一个 Figure 对象作为绘图区域。
-
创建一个 FigureCanvas 对象,将 Figure 对象绑定到画布上。
将画布添加到布局中。在绘图区域上创建一个子图(subplot)。 设置子图的横轴和纵轴标签。 设置子图的初始 x 轴范围为 0 到 20,y 轴范围为 0 到 100。 初始化绘图的线对象和均值线对象,并将它们加入子图中。 初始化存储数据的列表 x_data 和 y_data。 创建一个 FuncAnimation 动画对象,指定绘制的 Figure 对象、更新图形的函数 update_plot、总帧数为 200、更新间隔为 100 毫秒。
代码
def setup_ui(self):
# 创建中心部件和布局
self.central_widget = QWidget()
self.setCentralWidget(self.central_widget)
self.layout = QVBoxLayout(self.central_widget)
# 创建图形和画布
self.figure = Figure()
self.canvas = FigureCanvas(self.figure)
self.layout.addWidget(self.canvas)
# 添加子图和设置坐标轴
self.ax = self.figure.add_subplot(111)
self.ax.set_xlabel('时间')
self.ax.set_ylabel('随机数')
self.ax.set_xlim(0, 20)
self.ax.set_ylim(0, 100)
# 初始化绘图的线和均值线,以及数据存储
self.line, = self.ax.plot([], [], lw=2)
self.mean_line, = self.ax.plot([], [], color='red', lw=1)
self.x_data, self.y_data = [], []
# 创建动画对象,调用update_plot函数
self.anim = FuncAnimation(self.figure, self.update_plot, frames=200, interval=100)
update_plot函数
用于滑动窗口的显示与时刻更新
功能
基本逻辑为:
生成一个新的随机数。
(1)将新生成的随机数和对应的时间数据添加到 x_data 和 y_data 列表中。
(2)如果数据点数超过 200 个,则保持数据点数不超过 200 个,截取最新的 200 个数据点。
(3)使用 set_data 方法更新绘图线的数据。
(4)每隔 200 帧清空子图,重新设置子图的标签和范围,并重新创建绘图线对象和均值线对象。
(5)动态调整 x 轴的范围,使其在更新时始终显示最新的 20 个数据点。
(6)计算窗口内最新 20 个数据点的均值,并更新均值线的位置。
(7)刷新画布,以显示更新后的图形。
代码
def update_plot(self, i):
# 生成新的随机数
new_value = random.randint(0, 100)
# 更新时间和随机数数据
self.x_data.append(i * 0.1)
self.y_data.append(new_value)
# 保持数据点数不超过200个
if len(self.x_data) > 200:
self.x_data = self.x_data[-200:]
self.y_data = self.y_data[-200:]
# 更新绘图线
self.line.set_data(self.x_data, self.y_data)
# 每隔200帧清空图形
if i % 200 == 0:
self.ax.clear() # 清空子图
self.ax.set_xlabel('时间')
self.ax.set_ylabel('随机数')
self.ax.set_xlim(0, 20)
self.ax.set_ylim(0, 100)
self.line, = self.ax.plot([], [], lw=2)
self.mean_line, = self.ax.plot([], [], color='red', lw=1)
self.x_data, self.y_data = [], []
return self.line, self.mean_line
# 动态调整x轴范围
self.ax.set_xlim(max(0, (i - 20) * 0.1), max(2, i * 0.1))
# 计算窗口内数据的均值并更新均值线
window_data = self.y_data[-20:] # 当前窗口内的数据
window_mean = np.mean(window_data) if window_data else 0 # 计算均值
self.mean_line.set_data([self.x_data[-1] - 2, self.x_data[-1] + 2],
[window_mean, window_mean]) # 更新均值线
# 刷新画布
self.canvas.draw()
mian函数
def main():
app = QApplication(sys.argv)
window = MyMainWindow()
window.show()
sys.exit(app.exec_())