如何将多次迭代的多个时间序列数据绘制到pdf上

pip install matplotlib pandas reportlab

import pandas as pd
import matplotlib.pyplot as plt
from reportlab.pdfgen import canvas
from reportlab.lib.pagesizes import letter
from io import BytesIO

# 假设我们有一个包含多次迭代的时间序列数据的DataFrame
data = {
    'Iteration': ['Iter1', 'Iter2', 'Iter3'],
    'Time': ['2021-01-01', '2021-01-02', '2021-01-03'],
    'Value': [10, 15, 7]
}
df = pd.DataFrame(data)

# 创建一个PDF文件
pdf_path = 'time_series_data.pdf'
c = canvas.Canvas(pdf_path, pagesize=letter)

# 设置绘图区域
width, height = letter
margin = 50
plot_width = width - 2 * margin
plot_height = height - 2 * margin

# 绘制每个迭代的时间序列数据
for index, row in df.iterrows():
    time = row['Time']
    value = row['Value']
    
    # 创建一个内存中的图像
    buf = BytesIO()
    plt.figure(figsize=(plot_width / 100, plot_height / 100))
    plt.plot([time], [value], marker='o')
    plt.title(f'Iteration {row["Iteration"]}')
    plt.xlabel('Time')
    plt.ylabel('Value')
    plt.grid(True)
    
    # 保存图像到内存
    plt.savefig(buf, format='png')
    buf.seek(0)
    
    # 将图像绘制到PDF上
    c.drawImage(buf, margin, height - margin - plot_height, width=plot_width, height=plot_height)
    
    # 清除内存中的图像
    buf.close()
    plt.clf()

# 保存PDF文件
c.save()

print(f'PDF文件已生成：{pdf_path}')