plot_forward_error.py

from argparse import ArgumentParser
import tikzplotlib
import matplotlib
matplotlib.use('Agg')

import matplotlib.pyplot as plt  # noqa
from util import read_csv  # noqa

description = ("Plot the forward propagation errors")
parser = ArgumentParser(description=description)
parser.add_argument("--filename", type=str, required=True)
args = parser.parse_args()

filename = args.filename

results = read_csv(filename)

basename = "forward"

xvar = 'ntimesteps'

yvars = ['L0', 'L1', 'L2', 'Linf']

x_to_plot = results[xvar]

for yvar in yvars:
    # plt.xscale('log')
    # plt.yscale('log')
    plt.plot(x_to_plot, results[yvar])
    plt.xlabel(xvar)
    plt.ylabel(yvar)
    plt.savefig("%s_%s.pdf" % (basename, yvar), bbox_inches='tight')
    tikzplotlib.save("%s_%s.tex" % (basename, yvar))
    plt.clf()