Python作图-TXT文本

Posted 2022-12-15 thyhyh

# time:2022/11/29 10:09
import matplotlib.pyplot as plt
import numpy as np
#-----------  打开txt文件   ----------
file1 = open('D:/Net/epoch_miou.txt')
file2 = open('D:/Net/epoch_val_miou.txt')
# -----------  逐行读取文件内的数据  ------------
data1 = file1.readlines()
data2 = file2.readlines()
#------ x轴数据有序生成150个（根据自己的横坐标范围自己修改范围）  ----
x1 = np.arange(0,150)
#----------  新建一个空的列表，用于存储上一步逐行读取的data  ------------
y1 = []
y2 = []
#---------- 用循环的方式添加进列表  -----------
for num1 in data1:
    y1.append(float(num1.split(',')[0]))#TXT文本的第一列################################
for num2 in data2:
    y2.append(float(num2.split(',')[0]))#TXT文本的第一列################################
#---------------    输出图    ----------------------
#---------   可以理解为在图上加载x和y的数据   label为关于x和y曲线的标签------------
plt.plot(x1, y1, color='red',label='train mIoU')
plt.plot(x1, y2, color='grey',label='val mIoU')
# ---------   x轴的小标题   -------------
plt.xlabel('Epoch')
# ---------   y轴的小标题   -------------
plt.ylabel('mIoU')
plt.legend()
plt.show()