从数组返回最大值而不是数字值

Question

我有一个包含浮点值和NaN值的数组.我正在拿这个数组的中位数.在当前情景中,中位数可能是某个值或NaN.我的问题是,如果中位数是NaN,它应该从数组而不是NaN返回最大值.

数组示例,

x = np.array([NaN,NaN,0.1,0.002,0.14,NaN,NaN,NaN])

在这个数组中,np.median是NaN.当中位数是NaN然后它应该从数组返回最大值,我的意思是它应该从数组返回(0.14)值.

我怎样才能返回这个值？

Answer 1

import numpy as np
from numpy import NaN

x = np.array([NaN,NaN,0.1,0.002,0.14,NaN,NaN,NaN])

def getMedian(arr):
    med = np.median(arr)
    if np.isnan(med):
        return np.nanmax(arr)
    else:
        return med

print getMedian(x)