searchsorted

xarray_einstats.numba.searchsorted(da, v, dims=None, **kwargs)

Numbify numpy.searchsorted to support vectorized computations.

Parameters:

daxarray.DataArray

Input data

vxarray.DataArray

The values to insert into da.

dimsstr or iterable of str, optional

The dimensions over which to apply the searchsort. Computation will be parallelized over the rest with numba.

**kwargsdict, optional

Keyword arguments passed as-is to xarray.apply_ufunc.

Notes

It has been designed to be used by ecdf, so its setting of input and output core dims makes some assumptions based on that, it doesn’t aim to be general use vectorized/parallelized searchsorted.