Function nanquantile

dask/array/reductions.py:1750–1839

This works by automatically chunking the reduced axes to a single chunk and then calling ``numpy.nanquantile`` function across the remaining dimensions

(
    a,
    q,
    axis=None,
    out=None,
    overwrite_input=False,
    method="linear",
    keepdims=False,
    *,
    weights=None,
    interpolation=None,
)

@derived_from(np)
def nanquantile(
    a,
    q,
    axis=None,
    out=None,
    overwrite_input=False,
    method="linear",
    keepdims=False,
    *,
    weights=None,
    interpolation=None,
):
    """
    This works by automatically chunking the reduced axes to a single chunk
    and then calling ``numpy.nanquantile`` function across the remaining dimensions
    """
    if interpolation is not None:
        warnings.warn(
            "The `interpolation` argument to nanquantile was renamed to `method`.",
            FutureWarning,
            stacklevel=2,
        )

        if method != "linear":
            raise TypeError("Cannot pass interpolation and method keywords!")

        method = interpolation

    if axis is None:
        if builtins.any(n_blocks > 1 for n_blocks in a.numblocks):
            raise NotImplementedError(
                "The da.nanquantile function only works along an axis.  "
                "The full algorithm is difficult to do in parallel"
            )
        axis = tuple(range(len(a.shape)))
    elif not isinstance(axis, Iterable):
        axis = (axis,)
    axis = [ax + a.ndim if ax < 0 else ax for ax in axis]

    if overwrite_input:
        raise NotImplementedError(
            "da.nanquantile does not support overwrite_input=True"
        )

    # rechunk if reduced axes are not contained in a single chunk
    if builtins.any(a.numblocks[ax] > 1 for ax in axis):
        a = a.rechunk({ax: -1 if ax in axis else "auto" for ax in range(a.ndim)})

    q_arr = asarray_safe(q, like=a)
    if (
        HAS_NUMBAGG
        and (a.dtype.kind in "ui" or a.dtype == np.float64)
        and q_arr.dtype == np.float64
        and weights is None
        and method == "linear"
        and not keepdims
        and q_arr.ndim <= 1
    ):