diff --git a/modules/util/gfs_reader.py b/modules/util/gfs_reader.py index 098a071d46848dc289129edfc279358598f34080..6371062997b8ec5730d2f088393682eca63560a4 100644 --- a/modules/util/gfs_reader.py +++ b/modules/util/gfs_reader.py @@ -308,6 +308,21 @@ def get_time_interpolated_point(ds_0, ds_1, time0, time1, fld_name, time, lons, return intrp_vals +def get_time_interpolated_point_s(xr_dataset_s, time_s, fld_name_s, time, lons, lats, method='linear'): + pt_s = [] + for ds in xr_dataset_s: + pt = get_point_s(ds, fld_name_s, lons, lats) + pt_s.append(pt) + + pt = xr.concat(pt_s, 'time') + pt = pt.assign_coords(time=time_s) + + intrp_pt = pt.interp(time=time, method=method) + intrp_pt = intrp_pt.values + + return intrp_pt + + def get_time_interpolated_voxel(xr_dataset_s, time_s, time, fld_name, lon, lat, press, x_width=5, y_width=5, z_width=3, method='linear'): vox_s = [] for ds in xr_dataset_s: