xarray_broadcastable_from_dataset

PlasmaCalcs.tools.xarray_tools.xarray_dimensions.xarray_broadcastable_from_dataset(dataset, var=None, *, dims=None)

return dict of broadcastable versions of data_var(s) from dataset, standardizing data vars’ dims.

missing dims will be expanded to size 1. result’s dims will be put in order.

var: None, str, or list of str.

str –> return a broadcastable version of this data_var. Result is an xarray.DataArray.

list of str –> return dict of {v: broadcastable version of v} across v in var.

None –> use var = list(dataset.keys())

dims: None or list of str

result will have all these dims, in this order, with size 1 if not present in array.

if any array.dims not present in dims, raise DimensionKeyError.

None –> use dataset.dims.