jax.numpy.quantile

目录

jax.numpy.quantile#

jax.numpy.quantile(a, q, axis=None, out=None, overwrite_input=False, method='linear', keepdims=False, *, interpolation=Deprecated)[源代码][源代码]#

计算数据沿指定轴的分位数。

JAX 实现的 numpy.quantile()。

参数:

  • a (ArrayLike) – N 维数组输入。

  • q (ArrayLike) – 指定所需分位数的标量或一维数组。q 应包含介于 0.0 和 1.0 之间的浮点值。

  • axis (int | tuple[int, ...] | None) – 可选的轴或轴的元组，沿其计算分位数

  • out (None) – JAX 未实现；如果非 None 则会报错

  • overwrite_input (bool) – JAX 未实现此功能；如果未设置为 False，将会报错。

  • method (str) – 指定要使用的插值方法。选项是 ["linear", "lower", "higher", "midpoint", "nearest"] 之一。默认是 linear。

  • keepdims (bool) – 如果为真，则返回的数组将具有与输入相同的维度数量。默认为假。

  • interpolation (DeprecatedArg | str) – method 参数的已弃用别名。如果使用，将导致 DeprecationWarning。

返回:

包含指定轴上指定分位数的数组。

返回类型:

Array

参见

示例

计算数组的中位数和四分位数，使用线性插值：

>>> x = jnp.arange(10)
>>> q = jnp.array([0.25, 0.5, 0.75])
>>> jnp.quantile(x, q)
Array([2.25, 4.5 , 6.75], dtype=float32)

使用最近值插值计算四分位数：

>>> jnp.quantile(x, q, method='nearest')
Array([2., 4., 7.], dtype=float32)
目录