Skip to content
Snippets Groups Projects
Commit 1d964986 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 68c953e2
Branches
No related tags found
No related merge requests found
......@@ -443,7 +443,7 @@ def get_voxel_s(xr_dataset, fld_name_s, lon, lat, press, x_width=3, y_width=3, z
return sub_fld
def get_volume(xr_dataset, fld_name, press_range=None, lon_range=None, lat_range=None):
def get_volume(xr_dataset, fld_name, unit_str, press_range=None, lon_range=None, lat_range=None):
x_lo, x_hi = 0, NX
y_lo, y_hi = 0, NY
......@@ -477,6 +477,6 @@ def get_volume(xr_dataset, fld_name, press_range=None, lon_range=None, lat_range
sub_nda = nda[y_lo:y_hi, x_lo:x_hi, z_lo:z_hi]
xra = xr.DataArray(sub_nda, dims=['Latitude', 'Longitude', 'Pressure'],
coords={"Latitude": lat_coords[y_lo:y_hi], "Longitude": lon_coords[x_lo:x_hi], "Pressure": plevs[z_lo:z_hi]},
attrs={"description": fld_name, "units": 'm s-1'})
attrs={"description": fld_name, "units": unit_str})
return xra
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment