Skip to content

Instantly share code, notes, and snippets.

@darothen
Created October 12, 2015 23:58
Show Gist options
  • Save darothen/14ab2e03b0f1eb46a20b to your computer and use it in GitHub Desktop.
Save darothen/14ab2e03b0f1eb46a20b to your computer and use it in GitHub Desktop.
Convert xray.Dataarray to iris.Cube
def dataset_to_cube(ds, field):
""" Construct an iris Cube from a field of a given
xray DataSet. """
raise NotImplementedError("`iris` deprecated for Python 3")
dsf = ds[field]
## Attach coordinates to the cube, using full dataset for lookup
# dim_coords_and_dims = []
# for coord_key in dsf.coords:
# coord = ds[coord_key]
# coord_attrs = deepcopy(coord.attrs)
# idx_num = dsf.get_axis_num(coord_key)
#
# # Variable names
# standard_name, long_name, var_name = _get_dataset_names(ds,
# coord_key)
# # remove protected keys from attributes
# if long_name is not None:
# del coord_attrs['long_name']
# if standard_name is not None:
# del coord_attrs['standard_name']
#
# coord_units = _get_dataset_attr(coord, 'units')
# if coord_units is not None:
# del coord_attrs['units']
# else:
# coord_units = '1'
#
# coord_bounds = _get_dataset_attr(coord, 'bounds')
# if coord_bounds is not None:
# coord_bounds = ds[coord_attrs['bounds']]
# del coord_attrs['bounds']
#
# # Check if we need to coerce timestamps back to numerical
# # values
# if (
# ( coord_key == 'time' ) and
# ( 'datetime' in coord.dtype.name )
# ):
# points, unit, calendar = encode_cf_datetime(coord)
# coord_units = unit
# else:
# points = coord
#
# dc = DimCoord(points, standard_name=standard_name,
# long_name=long_name, var_name=var_name,
# units=coord_units, attributes=coord_attrs)
#
# dim_coords_and_dims.append( ( dc, idx_num ) )
#
# # Attach any attribute information to the cube
# attributes = deepcopy(dsf.attrs)
#
# standard_name, long_name, var_name = _get_dataset_names(ds, field)
# if long_name is not None:
# del attributes['long_name']
# if standard_name is not None:
# del attributes['standard_name']
#
# units = _get_dataset_attr(dsf, 'units')
# if units is not None:
# del attributes['units']
# else:
# units = '1'
#
# cell_methods = _get_dataset_attr(dsf, 'cell_methods')
# if cell_methods is not None:
# del attributes['cell_methods']
# cell_methods_list = []
# cell_methods = map(lambda s: s.strip(),
# cell_methods.split(":"))
# for coord, method in zip(cell_methods[::2], cell_methods[1::2]):
# cell_methods_list.append(
# CellMethod(method, coord)
# )
#
# cell_methods = cell_methods_list
#
# # Build / return the cube
# return Cube(dsf, standard_name=standard_name, long_name=long_name,
# var_name=var_name, units=units, attributes=attributes,
# cell_methods=cell_methods,
# dim_coords_and_dims=dim_coords_and_dims)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment