python - 在 python 中删除了 NaN 值的列表的中值

Question

是否可以在不显式删除 NaN 而忽略它们的情况下计算列表的中值？

我希望 median([1,2,3,NaN,NaN,NaN,NaN,NaN,NaN]) 为 2，而不是 NaN。

Answer 1

numpy 1.9.0 具有函数 nanmedian:

nanmedian(a, axis=None, out=None, overwrite_input=False, keepdims=False)
    Compute the median along the specified axis, while ignoring NaNs.

    Returns the median of the array elements.

    .. versionadded:: 1.9.0

例如

>>> from numpy import nanmedian, NaN
>>> nanmedian([1,2,3,NaN,NaN,NaN,NaN,NaN,NaN])
2.0

如果您不能使用 numpy 的 1.9.0 版，@Parker 的回答会起作用；例如

>>> import numpy as np
>>> x = np.array([1,2,3,NaN,NaN,NaN,NaN,NaN,NaN])
>>> np.median(x[~np.isnan(x)])
2.0

或

>>> np.median(x[np.isfinite(x)])
2.0

(当应用于 bool 数组时，~ 是 not 的一元运算符符号。)