Skip to content
Snippets Groups Projects
Commit 5118e5a6 authored by Ashwin Vishnu's avatar Ashwin Vishnu
Browse files

Load spatial means as an xarray dataset

parent 19b7531f
No related branches found
No related tags found
1 merge request!24Load spatial means as an xarray dataset
......@@ -103,6 +103,23 @@
dict_results = {}
return dict_results
def load_dataset(self, dims=("t",)):
"""Loads results as a xarray dataset."""
dict_results = self.load()
# NOTE: format specified in
# http://xarray.pydata.org/en/stable/generated/xarray.Dataset.from_dict.html
dset = {"coords": {}, "attrs": {}, "dims": dims, "data_vars": {}}
for key, value in dict_results.items():
if isinstance(value, np.ndarray):
target = "coords" if key in dims else "data_vars"
dset[target].update({key: {"dims": dims, "data": value}})
else:
dset["attrs"].update({key: value})
from xarray import Dataset
return Dataset.from_dict(dset)
def plot(self):
pass
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment