Updated script to filter NaN

MintonGroup · Sep 15, 2021 · e2e1cde · e2e1cde
1 parent 5e5ebd1
commit e2e1cde
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/examples/symba_mars_disk/aescattermovie.py b/examples/symba_mars_disk/aescattermovie.py
@@ -79,8 +79,7 @@ def setup_plot(self):
     def data_stream(self, frame=0):
         while True:
             d = self.ds.isel(time=frame)
-
-            d = d.where(d['radius'] < self.Rcb, drop=True)
+            d = d.where(np.invert(np.isnan(d['a'])), drop=True)
             d['radmarker'] = (d['radius'] / self.Rcb) * radscale
             radius = d['radmarker'].values