jax.scipy.stats.rankdata#

jax.scipy.stats.rankdata(a, method='average', *, axis=None, nan_policy='propagate')[原始碼]#

計算沿陣列軸的資料排名。

JAX 實作的 scipy.stats.rankdata()。

排名從 1 開始，而 method 參數控制如何處理平局。

參數:

  • a (ArrayLike) – 類陣列

  • method (str) – 字串，預設值=”average”。支援的方法為 ("average", "min", "max", "dense", "ordinal")。如需詳細資訊，請參閱 scipy.stats.rankdata() 文件。

  • axis (int | None) – 選用整數。如果未指定，則輸入陣列將被展平。

  • nan_policy (str) – 字串，JAX 的實作僅支援 "propagate"。

返回:

沿指定軸的排名陣列。

返回類型:

陣列 (Array)

範例

>>> x = jnp.array([10, 30, 20])
>>> rankdata(x)
Array([1., 3., 2.], dtype=float32)

>>> x = jnp.array([1, 3, 2, 3])
>>> rankdata(x)
Array([1. , 3.5, 2. , 3.5], dtype=float32)