fill missing values in MRD and MSLP

oceanmodeling · Mar 30, 2022 · ba2992a · ba2992a
1 parent 3bb1e03
commit ba2992a
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/stormevents/nhc/track.py b/stormevents/nhc/track.py
@@ -962,6 +962,25 @@ def unfiltered_data(self) -> DataFrame:
             dataframe.sort_values(['datetime', 'advisory'], inplace=True)
             dataframe.reset_index(inplace=True, drop=True)
 
+            # fill missing values of MRD and MSLP in the OFCL advisory
+            advisories = pandas.unique(dataframe['advisory'])
+            if 'OFCL' in advisories and 'CARQ' in advisories:
+                is_ofcl = dataframe['advisory'] == 'OFCL'
+                is_carq = dataframe['advisory'] == 'CARQ'
+
+                ofcl_mrd_missing = is_ofcl & pandas.isna(dataframe['radius_of_maximum_winds'])
+                ofcl_mslp_missing = is_ofcl & pandas.isna(dataframe['central_pressure'])
+
+                # fill OFCL maximum wind radius with the first entry from the CARQ advisory
+                dataframe.loc[ofcl_mrd_missing, 'radius_of_maximum_winds'] = dataframe.loc[
+                    is_carq, 'radius_of_maximum_winds'
+                ].iloc[0]
+
+                # fill OFCL central pressure (at sea level) with the first entry from the CARQ advisory
+                dataframe.loc[ofcl_mslp_missing, 'central_pressure'] = dataframe.loc[
+                    is_carq, 'central_pressure'
+                ].iloc[0]
+
             self.__unfiltered_data = dataframe
             self.__previous_configuration = configuration