jax.numpy.percentile#

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

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

numpy.percentile() 的 JAX 實作。

參數:

  • a (ArrayLike) – N 維陣列輸入。

  • q (ArrayLike) – 指定所需分位數的純量或 1 維陣列。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。

傳回:

一個陣列，包含沿指定軸的指定百分位數。

傳回類型:

Array

另請參閱

範例

計算 1 維陣列的中位數和四分位數

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

使用 nearest 而非線性內插計算相同的百分位數

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