jax.numpy.quantile#

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

沿著指定軸計算資料的分位數。

numpy.quantile() 的 JAX 實作。

參數:

  • 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) – 若為 True，則傳回的陣列將與輸入具有相同的維度數。預設值為 False。

  • 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)