Python 实时数据可视化教程：Matplotlib 与 Pygame 实践

import numpy as np
import matplotlib.pyplot as plt
import time # 用于模拟实时数据流

# 1. 创建图表和子图
fig, (ax1, ax2) = plt.subplots(2, 1, figsize=(10, 8))
fig.suptitle('实时温度与湿度数据', fontsize=16)

# 2. 初始化数据
# x轴数据通常是时间或数据点索引
x_data = np.arange(100)
# 初始y轴数据
y1_data = np.random.rand(100) * 30 + 20 # 模拟温度 (20-50)
y2_data = np.random.rand(100) * 40 + 40 # 模拟湿度 (40-80)

# 3. 绘制初始线条并获取Line2D对象
# 使用逗号解包，因为plot返回一个Line2D对象的列表
line1, = ax1.plot(x_data, y1_data, 'r-', label='温度')
line2, = ax2.plot(x_data, y2_data, 'b-', label='湿度')

# 4. 设置子图属性
ax1.set_title('温度')
ax1.set_ylim(0, 100)
ax1.set_xlim(0, 100) # 保持x轴范围固定，模拟滚动窗口
ax1.set_ylabel('摄氏度 (°C)')
ax1.legend()
ax1.grid(True)

ax2.set_title('湿度')
ax2.set_ylim(0, 100)
ax2.set_xlim(0, 100)
ax2.set_ylabel('相对湿度 (%)')
ax2.legend()
ax2.grid(True)

plt.tight_layout(rect=[0, 0.03, 1, 0.95]) # 调整布局，防止标题重叠

# 5. 开启交互模式并显示图表
plt.ion()
plt.show()

# 6. 实时数据更新循环
print("开始实时数据更新...")
for i in range(500): # 模拟更新500次
    # 模拟新的传感器读数
    # 基于上一个值加上随机噪声和一些趋势
    new_temp = np.clip(y1_data[-1] + (np.random.randn() * 1.5) + np.sin(i/20)*0.5, 0, 100)
    new_hum = np.clip(y2_data[-1] + (np.random.randn() * 1.0) + np.cos(i/15)*0.3, 0, 100)

    # 更新数据数组（滚动窗口效果）
    y1_data = np.append(y1_data[1:], new_temp)
    y2_data = np.append(y2_data[1:], new_hum)

    # 更新Line2D对象的数据
    line1.set_ydata(y1_data)
    line2.set_ydata(y2_data)

    # 7. 强制重绘并暂停
    plt.draw()
    plt.pause(0.01) # 短暂暂停，允许UI更新

    if i % 50 == 0:
        print(f"更新 {i+1} 次，当前温度: {new_temp:.2f}°C, 湿度: {new_hum:.2f}%")

print("实时数据更新结束。")
# 8. 关闭交互模式，并保持最终图表显示直到手动关闭
plt.ioff()
plt.show()

import pygame
import sys
import random
import math

# 初始化 Pygame
pygame.init()

# 屏幕尺寸
SCREEN_WIDTH = 1200
SCREEN_HEIGHT = 700
screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
pygame.display.set_caption("Pygame 实时图表示例")

# 颜色定义
WHITE = (255, 255, 255)
BLACK = (0, 0, 0)
RED = (255, 0, 0)
BLUE = (0, 0, 255)
GREEN = (0, 255, 0)

class Graph:
    def __init__(self, screen_width, screen_height, initial_list, top_value, bottom_value, color):
        """
        初始化图表对象。
        :param screen_width: 绘制区域的宽度
        :param screen_height: 绘制区域的高度
        :param initial_list: 初始数据列表
        :param top_value: 数据范围的上限（例如，温度的最高值）
        :param bottom_value: 数据范围的下限（例如，温度的最低值）
        :param color: 图表的颜色
        """
        self.screen_width = screen_width
        self.screen_height = screen_height
        self.color = color
        self.data_list = initial_list[:] # 复制一份数据，避免修改原始列表

        # 计算每个数据点的x轴单位宽度
        # 如果数据列表只有一个点，单位宽度设为屏幕宽度，防止除以零
        self.unit_x = self.screen_width / max(1, len(self.data_list) - 1)

        # 存储数据范围
        self.data_top = top_value
        self.data_bottom = bottom_value

    def _map_value_to_y(self, value):
        """
        将数据值映射到屏幕的y坐标。
        Pygame的y轴原点在左上角，向下递增。
        :param value: 原始数据值
        :return: 映射后的屏幕y坐标
        """
        # 防止除以零
        data_range = self.data_top - self.data_bottom
        if data_range == 0:
            return self.screen_height / 2 # 如果范围为0，则居中显示

        # 将值归一化到0-1，然后反转（因为y轴向下）
        normalized_value = (value - self.data_bottom) / data_range
        # 映射到屏幕高度，并从底部开始计算
        y = self.screen_height - (normalized_value * self.screen_height)
        return round(y)

    def update_data(self, new_data_point):
        """
        更新数据列表，实现滚动效果。
        :param new_data_point: 新的单个数据点
        """
        self.data_list.append(new_data_point)
        # 保持数据点数量不变，移除最旧的数据
        if len(self.data_list) > int(self.screen_width / self.unit_x) + 1: # 确保数据点覆盖整个宽度
            self.data_list.pop(0)

        # 重新计算x轴单位宽度，以防数据点数量变化
        self.unit_x = self.screen_width / max(1, len(self.data_list) - 1)


    def draw(self, surface):
        """
        在给定的surface上绘制图表。
        :param surface: Pygame的Surface对象，通常是主屏幕
        """
        if len(self.data_list) < 2:
            return # 至少需要两个点才能画线

        for i in range(len(self.data_list) - 1):
            x1 = i * self.unit_x
            y1 = self._map_value_to_y(self.data_list[i])

            x2 = (i + 1) * self.unit_x
            y2 = self._map_value_to_y(self.data_list[i+1])

            pygame.draw.line(surface, self.color, (x1, y1), (x2, y2), 2) # 线的粗细为2

# --- Pygame 主循环演示 ---
clock = pygame.time.Clock()

# 初始数据
initial_temp_data = [random.uniform(20, 30) for _ in range(100)]
initial_hum_data = [random.uniform(50, 70) for _ in range(100)]

# 创建温度和湿度图表实例
temp_graph = Graph(SCREEN_WIDTH, SCREEN_HEIGHT // 2 - 20, initial_temp_data, 100, 0, RED)
hum_graph = Graph(SCREEN_WIDTH, SCREEN_HEIGHT // 2 - 20, initial_hum_data, 100, 0, BLUE)

# 为了在屏幕上区分两个图表，调整它们的绘制区域
# 温度图在上方
temp_graph_rect = pygame.Rect(0, 0, SCREEN_WIDTH, SCREEN_HEIGHT // 2 - 10)
# 湿度图在下方
hum_graph_rect = pygame.Rect(0, SCREEN_HEIGHT // 2 + 10, SCREEN_WIDTH, SCREEN_HEIGHT // 2 - 10)

running = True
frame_count = 0
while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False

    # 模拟新数据点
    frame_count += 1
    new_temp = random.uniform(20, 30) + math