domino.agg

 1from domino.categorical_analysis import get_xr_occurrence
 2
 3def aggregate_ds(ds,dim,agg_func,**agg_kwargs):
 4    """
 5    A wrapper function for applying aggregation functions to datasets.
 6    **Arguments**
 7    *ds* An xarray.Dataset to aggregate
 8    *dim* A string specifiying the dimension of *ds* to aggregate over.
 9    *agg_func* An aggregation function with signature (xarray.Dataset,string,**agg_kwargs)->(xarray.Dataset)
10    
11    **Outputs**
12    An xarray.Dataset aggregated over *dim*.
13    """
14    return ds.map(agg_func,dim=dim,**agg_kwargs)
15def mean_ds(ds,dim):
16    """Returns the mean of the dataset over dim"""
17    return aggregate_ds(ds,dim,lambda da,dim: da.mean(dim=dim))
18def std_ds(ds,dim):
19    """Returns the standard deviation of the dataset over dim"""
20    return aggregate_ds(ds,dim,lambda da,dim: da.std(dim=dim))
21def cat_occ_ds(ds,dim,cat_ds):
22    """Returns the occurrence frequency of the dataset's categorical variables"""
23    return aggregate_ds(ds,dim,get_xr_occurrence,s=cat_ds,coord_name='variable_cat_val')