radialV: mask empty bins

MPoL-dev · Oct 27, 2023 · 10b7fb2 · 10b7fb2
1 parent c009f23
commit 10b7fb2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/mpol/onedim.py b/src/mpol/onedim.py
@@ -152,6 +152,10 @@ def radialV(V, coords, geom, rescale_flux, bins=None):
     # cumulative binned visibility amplitude in each annulus
     Vs, _ = np.histogram(a=qq, bins=bins, weights=Vp)
 
+    # mask empty bins
+    mask = (bin_counts == 0)
+    Vs = np.ma.masked_where(mask, Vs)
+
     # average binned visibility amplitude in each annulus
     Vs /= bin_counts