txt文档中保存的数据如下:
绘制图像的代码如下:
import matplotlib.pyplot as plt
file = open('/Users/two8g/shao_work/loss.txt') # 打开文档
data = file.readlines() # 读取文档数据
print(data)
para_1 = [] # 新建列表,用于保存第一列数据
para_2 = [] # 新建列表,用于保存第二列数据
for num in data:
try:
temp = num.split(':')
t = temp[1].split(',')
except:
continue
# print(num.split(':')[1].split(' ')[0])
para_1.append(float(num.split(':')[1].split(' ')[0]))
para_2.append(float(num.split(':')[2]))
# print(para_1)
plt.figure()
plt.title('loss')
plt.plot(para_1, para_2)
plt.show()
绘制的图像如下: