scipy for xarray

xr-scipy is a thin wrapper of scipy for the xarray eco-system.

Many scipy functions, such as scipy.integrate.trapz requires coordinate array as an argument. xr-scipy wraps these functions to use native coordinate objects of xarray and returns an xarray object with the computed data. This enables more xarray-oriented data analysis with scipy.

Other usage/options are kept almost the same to the original scipy function.

Example

In [1]: da = xr.DataArray([0, 3, 2, 4, 6], dims='x',
   ...:                       coords={'x': np.linspace(0, 1, 5)})
In [2]: da
Out[2]:
<xarray.DataArray (x: 5)>
array([0, 3, 2, 4, 6])
Coordinates:
  * x        (x) float64 0.0 0.25 0.5 0.75 1.0

In [3]: xrscipy.integrate.cumtrapz(da, coord='x')
Out[3]:
<xarray.DataArray (x: 5)>
array([0.   , 0.375, 1.   , 1.75 , 3.   ])
Coordinates:
  * x        (x) float64 0.0 0.25 0.5 0.75 1.0