Add hour selection option

Author: DamienIrving

adjust.py

@@ -227,6 +227,7 @@ def main(args):
         args.infiles,
         args.var,
         time_bounds=args.adjustment_tbounds,
+        isel_hour=args.isel_hour,
         input_units=args.input_units,
         output_units=args.output_units,
         use_cftime=False,
@@ -288,6 +289,12 @@ def main(args):
 
     parser.add_argument("--input_units", type=str, default=None, help="input data units")
     parser.add_argument("--output_units", type=str, default=None, help="output data units")
+    parser.add_argument(
+        "--isel_hour",
+        type=int,
+        default=None,
+        help="select a single hour from the infiles"
+    )
     parser.add_argument(
         "--adjustment_tbounds",
         type=str,

train.py

@@ -125,6 +125,7 @@ def main(args):
         args.hist_files,
         args.hist_var,
         time_bounds=args.hist_time_bounds,
+        isel_hour=args.isel_hour,
         input_units=args.input_hist_units,
         output_units=args.output_units,
         valid_min=args.valid_min,
@@ -135,6 +136,7 @@ def main(args):
         args.ref_files,
         args.ref_var,
         time_bounds=args.ref_time_bounds,
+        isel_hour=args.isel_hour,
         lat_bounds=args.lat_bounds,
         lon_bounds=args.lon_bounds,
         input_units=args.input_ref_units,
@@ -196,6 +198,12 @@ def main(args):
         required=True,
         help="reference data files"
     )
+    parser.add_argument(
+        "--isel_hour",
+        type=int,
+        default=None,
+        help="select a single hour from the input files"
+    )
     parser.add_argument(
         "--hist_time_bounds",
         type=str,

utils.py

@@ -204,6 +204,7 @@ def read_data(
     time_bounds=None,
     lat_bounds=None,
     lon_bounds=None,
+    isel_hour=None,
     input_units=None,
     output_units=None,
     lon_chunk_size=None,
@@ -225,6 +226,8 @@ def read_data(
         Rename var to value of rename_var
     time_bounds : list, optional
         Time period to extract from infiles [YYYY-MM-DD, YYYY-MM-DD]
+    isel_hour : int, optional
+        Select a single hour from the infiles
     lat_bnds : list, optional
         Latitude bounds: [south bound, north bound] 
     lon_bnds : list, optional
@@ -278,7 +281,11 @@ def read_data(
 
     if time_bounds:
         start_date, end_date = time_bounds
-        ds = ds.sel({'time': slice(start_date, end_date)})        
+        ds = ds.sel({'time': slice(start_date, end_date)})
+
+    if type(isel_hour) == int:
+        ds = ds.isel(time=(ds.time.dt.hour == isel_hour))
+
     if lat_bounds:
         ds = subset_lat(ds, lat_bounds)
     if lon_bounds: