填充之众数填充

# 定义fill_category_na函数
def fill_category_na(df, columns_list):
    '''
    该函数功能是将样本数据中所有类别型特征中的缺失数据填充为该特征的众数
    df:输入的数据集
    columns_list:为一个列表集合,里面的元素为每个具有缺失值的类别型特征
    最后函数返回值为将缺失值填充成相应特征平均值的样本数据
    '''
    # 遍历每一个具有缺失值的类别型特征
    for column in columns_list:
        # 对每一个类别型特征计算出每个类别的数量
        counts = df[column].value_counts()
        # 将该特征的众数作为填充值 
        df[column].fillna(np.argmax(counts), axis=0, inplace=True)
    # 函数返回值为填充好的样本数据
    return df
posted @ 2020-03-17 21:18  宋桓公  阅读(29)  评论(0编辑  收藏  举报