Fix issue where sep/pa astrometry was being plotted as ra/dec

sblunt · Aug 12, 2024 · f3ba63f · f3ba63f
1 parent 55ab594
commit f3ba63f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/orbitize/system.py b/orbitize/system.py
@@ -755,14 +755,15 @@ def plot_astrometry(self, center_on_star=False):
 
             # get radec astrometry
             for i in range(n_obs):
-                if np.isin(i, self.all_radec): # get radec astrometry
+                j = np.where(mask_obj)[0][i] # index obj entries in input table
+                if np.isin(j, self.all_radec): # get radec astrometry
                     ra[i] = data_table_obj[i]['quant1']
                     ra_err[i] = data_table_obj[i]['quant1_err']
                     dec[i] = data_table_obj[i]['quant2']
                     dec_err[i] = data_table_obj[i]['quant2_err']
                     radec_corr[i] = data_table_obj[i]['quant12_corr']
 
-                elif np.isin(i, self.all_seppa): # get seppa astrometry, convert to radec
+                elif np.isin(j, self.all_seppa): # get seppa astrometry, convert to radec
                     sep = data_table_obj[i]['quant1']
                     sep_err = data_table_obj[i]['quant1_err']
                     pa = data_table_obj[i]['quant2']