"""Module to add a data variable to a dataset."""
import numpy as np
[docs]
def add_data_var(ds_id, ds, **operands):
"""
Add a data variable.
Parameters
----------
ds_id : str
Dataset ID.
Unused in this function.
ds : xarray.Dataset
A Dataset.
operands : dict
Dictionary containing the new data variable.
Returns
-------
xarray.Dataset
"""
var_id = operands.get("var_id")
value = operands.get("value")
dtype = operands.get("dtype")
ds = ds.assign({f"{var_id}": np.array(value, dtype=dtype)})
for k, v in operands.get("attrs").items():
ds[var_id].attrs[k] = v
return ds