我们有 pandas 的 fillna 来填补缺失值。
让我们通过示例数据框了解一些用例:
df = pd.DataFrame({'col1':['John', np.nan, 'Anne'], 'col2':[np.nan, 3, 4]})
col1 col2
0 John NaN
1 NaN 3.0
2 Anne 4.0
如文档中所述,fillna 接受以下内容作为填充 values:
值:标量、字典、系列或数据帧
所以我们可以用一个常量值替换,比如一个空字符串:
df.fillna('')
col1 col2
0 John
1 3
2 Anne 4
1
您也可以替换为字典映射column_name:replace_value:
df.fillna({'col1':'Alex', 'col2':2})
col1 col2
0 John 2.0
1 Alex 3.0
2 Anne 4.0
或者您也可以替换为另一个 pd.Series 或 pd.DataFrame:
df_other = pd.DataFrame({'col1':['John', 'Franc', 'Anne'], 'col2':[5, 3, 4]})
df.fillna(df_other)
col1 col2
0 John 5.0
1 Franc 3.0
2 Anne 4.0
这非常有用,因为它允许您使用从列中提取的一些统计信息来填充数据框列上的缺失值,例如 mean 或 mode。假设我们有:
df = pd.DataFrame(np.random.choice(np.r_[np.nan, np.arange(3)], (3,5)))
print(df)
0 1 2 3 4
0 NaN NaN 0.0 1.0 2.0
1 NaN 2.0 NaN 2.0 1.0
2 1.0 1.0 2.0 NaN NaN
那么我们就可以轻松做到了:
df.fillna(df.mean())
0 1 2 3 4
0 1.0 1.5 0.0 1.0 2.0
1 1.0 2.0 1.0 2.0 1.0
2 1.0 1.0 2.0 1.5 1.5