Project-1/src/plot_general_alg_error.py
2023-09-12 19:09:12 +02:00

29 lines
1013 B
Python

import matplotlib.pyplot as plt
def main():
fig_a, ax_a = plt.subplots()
fig_b, ax_b = plt.subplots()
# fig_c, ax_c = plt.subplots()
for i in range(6):
x = []
abs_err = []
rel_err = []
with open(f"output/error/out_{10**(i+1)}.txt", "r") as f:
lines = f.readlines()
for line in lines:
x_i, abs_err_i, rel_err_i = line.strip().split(",")
x.append(float(x_i))
abs_err.append(float(abs_err_i))
rel_err.append(float(rel_err_i))
ax_a.plot(x, abs_err, label=f"abs_err {10**(i+1)} steps")
ax_b.plot(x, rel_err, label=f"rel_err {10**(i+1)} steps")
ax_a.legend(bbox_to_anchor=(1.05, 1), loc=2, borderaxespad=0.)
ax_b.legend(bbox_to_anchor=(1.05, 1), loc=2, borderaxespad=0.)
fig_a.savefig("../latex/images/problem8_a.pdf", bbox_inches="tight")
fig_b.savefig("../latex/images/problem8_b.pdf", bbox_inches="tight")
if __name__ == "__main__":
main()