Skip to content
Snippets Groups Projects
Commit 429a7da5 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 88a28731
No related merge requests found
......@@ -267,7 +267,7 @@ def get_point_s(xr_dataset, fld_name_s, lons, lats, method='nearest'):
def get_time_interpolated_vert_profile(xr_dataset_s, time_s, fld_name, time, lons, lats, method='linear'):
prof_s = []
for ds in xr_dataset_s:
vp = get_vert_profile(ds, fld_name, lons, lats)
vp = get_vert_profile(ds, fld_name, lons, lats, method=method)
prof_s.append(vp)
prof = xr.concat(prof_s, 'time')
......@@ -282,7 +282,7 @@ def get_time_interpolated_vert_profile(xr_dataset_s, time_s, fld_name, time, lon
def get_time_interpolated_vert_profile_s(xr_dataset_s, time_s, fld_name_s, time, lons, lats, method='linear'):
prof_s = []
for ds in xr_dataset_s:
vp = get_vert_profile_s(ds, fld_name_s, lons, lats)
vp = get_vert_profile_s(ds, fld_name_s, lons, lats, method=method)
prof_s.append(vp)
prof = xr.concat(prof_s, 'time')
......@@ -294,14 +294,14 @@ def get_time_interpolated_vert_profile_s(xr_dataset_s, time_s, fld_name_s, time,
return intrp_prof
def get_time_interpolated_point(ds_0, ds_1, time0, time1, fld_name, time, lons, lats):
def get_time_interpolated_point(ds_0, ds_1, time0, time1, fld_name, time, lons, lats, method='linear'):
vals_0 = get_point(ds_0, fld_name, lons, lats)
vals_1 = get_point(ds_1, fld_name, lons, lats)
vals = xr.concat([vals_0, vals_1], 'time')
vals = vals.assign_coords(time=[time0, time1])
intrp_vals = vals.interp(time=time, method='linear')
intrp_vals = vals.interp(time=time, method=method)
intrp_vals = intrp_vals.values
......@@ -311,7 +311,7 @@ def get_time_interpolated_point(ds_0, ds_1, time0, time1, fld_name, time, lons,
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 = get_point_s(ds, fld_name_s, lons, lats, method=method)
pt_s.append(pt)
pt = xr.concat(pt_s, 'time')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment