jax.numpy.nanmedian

目录

jax.numpy.nanmedian#

jax.numpy.nanmedian(a, axis=None, out=None, overwrite_input=False, keepdims=False)[源代码][源代码]#

返回沿给定轴的数组元素的中位数，忽略 NaNs。

JAX 实现的 numpy.nanmedian()。

参数:

  • a (ArrayLike) – 输入数组。

  • axis (int | tuple[int, ...] | None) – 可选，整数或整数序列，默认=None。计算中位数的轴。如果为None，则对展平的数组计算中位数。

  • keepdims (bool) – bool, 默认=False。如果为真，缩减的轴将保留在结果中，尺寸为1。

  • out (None) – 未被 JAX 使用。

  • overwrite_input (bool) – 未被 JAX 使用。

返回:

包含沿指定轴的中位数，忽略 NaNs 的数组。如果沿指定轴的所有元素都是 NaNs，则返回 nan。

返回类型:

Array

参见

示例

默认情况下，中位数是针对展平的数组计算的。

>>> nan = jnp.nan
>>> x = jnp.array([[2, nan, 7, nan],
...                [nan, 5, 9, 2],
...                [6, 1, nan, 3]])
>>> jnp.nanmedian(x)
Array(4., dtype=float32)

如果 axis=1，则沿轴 1 计算中位数。

>>> jnp.nanmedian(x, axis=1)
Array([4.5, 5. , 3. ], dtype=float32)

如果 keepdims=True，输出的 ndim 等于输入的 ndim。

>>> jnp.nanmedian(x, axis=1, keepdims=True)
Array([[4.5],
       [5. ],
       [3. ]], dtype=float32)
目录