python matplotlib 여러개 그래프 그리기

문제

• plot을 여러개 그리고 싶은데 한꺼번에 돌리게 되면 plot이 하나만 그려지거나 이상하게 그려지는 경우가 발생

해결

• plt.figure('인자') 를 넣어주면 한꺼번에 돌려도 각각 따로 plot이 그려지게 됨.

 

    plt.figure(1)
    plt.plot(epoch, trn_loss1, '-b', label='train_loss')
    plt.plot(epoch, trn_loss2, '-r', label='val_loss')

    plt.xlabel('Epoch')
    plt.legend(loc='upper right')
    plt.title('compare_loss')

    plt.savefig('경로.png')

    plt.figure(2)
    plt.plot(epoch, iou1, '-b', label='train-IoU')
    plt.plot(epoch, iou2, '-r', label='val-IoU')


    plt.xlabel('Epoch')
    plt.legend(loc='lower right')
    plt.title('compare_iou')
    plt.savefig('경로.png')