jax.numpy.percentile

内容

jax.numpy.percentile#

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

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

JAX 实现的 numpy.percentile().

参数:

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

  • q (ArrayLike) – 标量或一维数组，指定所需的百分位数。 q 应包含介于 0 和 100 之间的整数或浮点数。

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

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

  • overwrite_input (bool) – JAX 未实现；如果非 False，则会报错

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

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

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

返回:

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

返回类型:

数组

另请参见

示例

计算一维数组的中位数和四分位数

>>> x = jnp.array([0, 1, 2, 3, 4, 5, 6])
>>> q = jnp.array([25, 50, 75])
>>> jnp.percentile(x, q)
Array([1.5, 3. , 4.5], dtype=float32)

使用最近插值而不是线性插值计算相同的百分位数

>>> jnp.percentile(x, q, method='nearest')
Array([1., 3., 4.], dtype=float32)
内容