如何在Matplotlib中实时动画展示多文件更新的大量数据？

我来帮你搞定这个持续绘制更新数据的需求！针对你这种多文件自动更新、每条文件对应一条Matplotlib曲线的场景，我整理了一套实用的方案，分步骤来实现：

首先得把每个文件里的迭代数和温度数据提取出来。你的文件开头有两行描述文本，后面是成对的数值，所以写一个专门的解析函数来处理：

import numpy as np

def parse_convergence_file(file_path):
    try:
        # 跳过前两行标题，读取后续数据
        data = np.loadtxt(file_path, skiprows=2)
        if data.ndim == 1:
            # 处理只有一组数据的情况（避免报错）
            iterations = [data[0]]
            temps = [data[1]]
        else:
            iterations = data[:, 0]
            temps = data[:, 1]
        return iterations, temps
    except OSError:
        # 文件正在被其他软件写入，暂时跳过
        return None, None

接下来要实现定期扫描文件、更新曲线的逻辑。我们可以用Matplotlib的交互式模式来实时刷新图表，同时跟踪每个文件的修改时间，避免重复读取未更新的文件：

import os
import matplotlib.pyplot as plt
import time

# 配置参数
TARGET_DIR = "./your_data_directory"  # 替换成你的文件目录
CHECK_INTERVAL = 2  # 每隔2秒扫描一次文件
FILE_PATTERN = "Convergence history"  # 匹配文件名的关键词

# 初始化绘图
plt.ion()
fig, ax = plt.subplots(figsize=(10, 6))
ax.set_xlabel("Iteration")
ax.set_ylabel("Vertex Average Static Temperature (SI)")
ax.set_title("Real-Time Convergence History")
ax.grid(True)

# 存储文件信息：键是文件路径，值是(修改时间, 曲线对象)
file_trackers = {}

while True:
    # 扫描目录下的目标文件
    for filename in os.listdir(TARGET_DIR):
        if FILE_PATTERN in filename:
            file_path = os.path.join(TARGET_DIR, filename)
            current_mtime = os.path.getmtime(file_path)
            
            # 处理新文件
            if file_path not in file_trackers:
                iterations, temps = parse_convergence_file(file_path)
                if iterations is not None:
                    # 添加新曲线，用文件名作为图例
                    line, = ax.plot(iterations, temps, label=filename)
                    file_trackers[file_path] = (current_mtime, line)
                    ax.legend()
            # 处理已更新的文件
            else:
                last_mtime, line = file_trackers[file_path]
                if current_mtime > last_mtime:
                    iterations, temps = parse_convergence_file(file_path)
                    if iterations is not None:
                        # 更新曲线数据
                        line.set_xdata(iterations)
                        line.set_ydata(temps)
                        file_trackers[file_path] = (current_mtime, line)
    
    # 自动调整坐标轴范围，避免数据超出显示
    ax.relim()
    ax.autoscale_view()
    # 刷新图表
    fig.canvas.draw()
    fig.canvas.flush_events()
    # 等待指定间隔
    time.sleep(CHECK_INTERVAL)

• 文件写入冲突处理：用try-except OSError捕获文件被占用的情况，跳过当前读取，下次循环再尝试，避免程序崩溃。
• 高效更新：通过跟踪文件的修改时间（os.path.getmtime），只处理有更新的文件，减少不必要的IO操作。
• 动态图例与坐标轴：新文件自动添加图例，每次更新后自动调整坐标轴范围，保证数据始终可见。
• 可定制参数：你可以根据需求调整CHECK_INTERVAL（扫描间隔）、TARGET_DIR（文件目录）和FILE_PATTERN（文件匹配规则）。

内容的提问来源于stack exchange，提问作者diegoiva