Python读取txt文件并作图保存

1.首先打开txt文件，观察txt文件中数据的属性以及数据之间的分隔符

例如上面的txt文件，只有一列数据，将其视为y轴数据，txt文件是按行存储，这就导致了等会我们每次按行读取数据的时候，每个数据后面会出现换行符'\n'，这就导致了我们读取的每个数据是str类型，即使只利用strip()方法去掉换行符，数据依然为str类型，所以在进行数据处理时，强制转换为float类型。

2.代码展示

from matplotlib import pyplot as plt
# 打开文本文件
with open('avg_valid_losses.txt', 'r') as file:
    # 读取文件内容
    lines = file.readlines()   #list类型

# 提取avg_train_losses数据
avg_valid_losses = []
for line in lines:
    avg_valid_losses.append(float(line.strip()))
    #avg_train_losses.append(line)
    # avg_train_losses.append(float(line))
# 绘制图形
plt.plot(avg_valid_losses, label='train_losses')
plt.savefig('valid_losses_plot.png')
# 显示图形
plt.show()