python - 创建一个自动填充空值的函数

Question

我当前的项目有大量带有空值的列。从技术上讲，我可以将它们一一填充或丢弃。但我想，也许我可以编写一个自动为我完成 fillna 的函数。这是我到目前为止所想到的，

def fill_null(df, column):
    if np.dtype(df[column]) == 'O':
       df[column].fillna('NA', inplace = True)
    elif np.dtype(df[column]) != 'O':
       df[column].fillna(0, inplace = True) 
    return

它有效。但是有没有办法让它自动迭代整个数据框并自动填充它们呢？或者我要求太多了。我对编程还是新手，不太擅长迭代。

谢谢!

Answer 1

使用DataFrame.select_dtypes对于对象列并创建字典以替换为 DataFrame.fillna ，然后将所有其他列替换为 0:

d = dict.fromkeys(df.select_dtypes(object).columns, 'NA')
df = df.fillna(d).fillna(0)

另一个解决方案:

df = df.apply(lambda x: x.fillna('NA') if x.dtype == 'O' else x.fillna(0))