用于替换NaN值的函数

Question

我试图用以下代码替换Nan值：

def NAfunc(df,s):

    if df[s].isnull():
        df[s]=df[s].fillna(df[s].mean())
    else:
        pass
    return df[s]
data['ColX']=data.apply(lambda x: NAfunc(x, 'ColX'), axis=1) #applying func to missing value column

我得到一个错误：

'float' object has no attribute 'isnull'

你能告诉我哪里出了问题，或者是一个更好的函数代码吗

Answer 1

不使用df[s].isnull()，您可以选择：

import math

def NAfunc(df,s):

    if math.isnan(df[s]):
        df[s]=df[s].fillna(df[s].mean())
    else:
        pass
    return df[s]