Update polar_analysis.py

Fixed ylim scaling issue in plot_background_fits
py4dstem · Aug 29, 2024 · 9867aa8 · 9867aa8
1 parent 72195a4
commit 9867aa8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py4DSTEM/process/polar/polar_analysis.py b/py4DSTEM/process/polar/polar_analysis.py
@@ -526,8 +526,8 @@ def plot_background_fits(
     ax.set_ylabel("I(k) and Background Fit Estimates")
     ax.set_ylim(
         (
-            np.min(self.radial_mean[self.radial_mean > 0]) * 0.8,
-            np.max(self.radial_mean * self.Sk_mask) * 1.25,
+            np.min(Ik[Ik > 0]) * 0.8,
+            np.max(Ik * self.Sk_mask) * 1.25,
         )
     )
     ax.set_yscale("log")