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
E.g.
>>> from numpy import nanmedian, NaN
>>> nanmedian([1,2,3,NaN,NaN,NaN,NaN,NaN,NaN])
2.0
如果你不能使用 numpy 1.9.0 版本,类似 @Parker's 的答案将会起作用;例如
>>> import numpy as np
>>> x = np.array([1,2,3,NaN,NaN,NaN,NaN,NaN,NaN])
>>> np.median(x[~np.isnan(x)])
2.0
or
>>> np.median(x[np.isfinite(x)])
2.0
(当应用于布尔数组时,~
是一元运算符表示法not
.)