plt画折线图

def display(valid_dict, test_dict):

    import matplotlib.pyplot as plt

    import time

    print('valid_dict', valid_dict)

    print('test_dict', test_dict)

    keys = list(valid_dict.keys())

    x_len = len(valid_dict[keys[0]])

    x=list(range(x_len))

    for key in keys:

        plt.figure()

        plt.plot(x, valid_dict[key], 'r-x', label='valid_'+key)  # 画线并添加图例legend

        plt.plot(x, test_dict[key], 'g-^', label='test_'+key)  # 画线并添加图例legend

        plt.legend()  # 展示图例legend

        plt.xlabel('epoch')  # 给 x 轴添加坐标轴信息

        plt.ylabel(key)  # 给 y 轴添加坐标轴信息

        plt.title('test&valid'+key)  # 添加图片标题

        plt.axis('on')   # 关闭坐标轴的显示

        name = key+ '-'+ time.strftime('%Y-%m-%d-%H:%M:%S',time.localtime())+'.jpg'

        plt.savefig(name)

        plt.show()