clean up plot

peterdsharpe · Jan 30, 2024 · 03d5c5f · 03d5c5f
1 parent 26f74df
commit 03d5c5f
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/...integrate_discrete_derivations/discrete_squared_curvature_display_spectral_convergence.py b/...integrate_discrete_derivations/discrete_squared_curvature_display_spectral_convergence.py
@@ -14,7 +14,7 @@
 print(float(exact))
 
 def get_approx(period=10, method="cubic"):
-    x_vals = np.cosspace(0, 1, n_samples).astype(float)
+    x_vals = np.linspace(0, 1, n_samples).astype(float)
     f_vals = s.lambdify(
         x,
         f.subs(k, (n_samples - 1) / period),
@@ -40,8 +40,8 @@ def get_approx(period=10, method="cubic"):
 
 import matplotlib.pyplot as plt
 import aerosandbox.tools.pretty_plots as p
-fig, ax = plt.subplots(2, 1)
-for method in ["cubic", "simpson", "subgradient", "hybrid_simpson_cubic"]:
+fig, ax = plt.subplots(2, 1, figsize=(6, 8))
+for method in ["cubic", "simpson", "hybrid_simpson_cubic"]:
     approxes = np.vectorize(
         lambda period: get_approx(period, method=method)
     )(periods)
@@ -51,4 +51,13 @@ def get_approx(period=10, method="cubic"):
 plt.xlabel("Period [samples]")
 ax[0].set_ylabel("Relative Error")
 ax[1].set_ylabel("Approximation")
+plt.sca(ax[1])
+p.hline(
+    float(exact),
+    color="k",
+    linestyle="--",
+    label="Exact",
+    alpha=0.5
+)
+
 p.show_plot()