plt画图代码

plt画图代码

# 读取文件中的数据，并画图
# 改文件代码使用的前提条件：数据格式要和该文件的数据格式保持一样
import matplotlib.pyplot as plt


def file_data(file):
    data = []
    file_data = file.readlines()  # 读取所有行
    for row in file_data:
        row.replace('\n', '')
        data.append(round((float(row)), 2))  # 将每行数据插入data中，round：保留小数操作
    return data


def plt_drwing(X_numbers: int, Y_numbers: int, Y_ticks_Max=10, Y_interval=1, Y_limit=2, Legend_name='Loss',
               X_label_name='Epochs', Y_label_name='Numbers', Title_name='Loss'):
    X_numbers = list(range(1, X_numbers + 1))
    plt.figure(figsize=(15, 15))  # 设置画布大小
    plt.tick_params(labelsize=20)  # 刻度字体大小13
    plt.plot(X_numbers, Y_numbers, color='r')  # 画图，并设置图线的颜色
    # plt.plot(Z_numbers, W_numbers, color='B')  # 当需要在一张图上面画多条曲线时，可以添加该条代码
    plt.yticks(range(0, Y_ticks_Max, Y_interval))  # y轴数据显示数据
    plt.ylim((0, Y_limit))  # y轴数据显示限制

    plt.legend([Legend_name], fontsize=20)  # 当一幅图画了多条线时，需要添加多个图例时，可以参照该条代码
    # plt.legend([Legend_name01, Legend_name01], fontsize=20) # 当一幅图画了多条线时，需要添加多个图例时，可以参照该条代码
    plt.xlabel(X_label_name, fontsize=20)
    plt.ylabel(Y_label_name, fontsize=20)
    plt.title(Title_name, fontsize=20)
    plt.show()


def main():
    file01_acc = open('./test_acc.txt', 'r')  # 打开文件
    Y_nubmers = file_data(file01_acc)
    X_numbers = len(Y_nubmers)
    plt_drwing(X_numbers, Y_nubmers)


if __name__ == '__main__':
    main()

处理的数据格式是这样的：