如何通过比较从python中的两个不同文本文件读取的数据来生成绘图？

import matplotlib.pyplot as plt

# 读取第一个文件的数据
with open('file1.txt', 'r') as f1:
    data1 = [float(line.strip()) for line in f1]

# 读取第二个文件的数据
with open('file2.txt', 'r') as f2:
    data2 = [float(line.strip()) for line in f2]

# 确保两个数据集的长度相同
assert len(data1) == len(data2), "数据长度不一致"

# 生成x轴坐标（假设每行数据对应一个时间点）
x = range(len(data1))

# 绘制折线图
plt.figure(figsize=(10, 5))
plt.plot(x, data1, label='File 1 Data', marker='o')
plt.plot(x, data2, label='File 2 Data', marker='s')

# 添加图例和标签
plt.legend()
plt.xlabel('Index')
plt.ylabel('Value')
plt.title('Comparison of Data from Two Files')

# 显示图表
plt.show()