时间戳

首页 > 解决方案 > 从同一组的其他记录中的值替换熊猫数据框中的 NAN 值

python - 从同一组的其他记录中的值替换熊猫数据框中的 NAN 值

问题描述

我有一个数据框df 

import pandas as pd
import numpy as np

df = pd.DataFrame({'A': [np.nan, 1, 2,np.nan,2,np.nan,np.nan], 
               'B': [10, np.nan, np.nan,5,np.nan,np.nan,7], 
               'C': [1,1,2,2,3,3,3]})

看起来像：

     A     B  C
0  NaN  10.0  1
1  1.0   NaN  1
2  2.0   NaN  2
3  NaN   5.0  2
4  2.0   NaN  3
5  NaN   NaN  3
6  NaN   7.0  3

我想替换 column 中的所有 NAN 值A以及B来自与 column 中提到的同一组的其他记录中的值C。

我的预期输出是：

     A     B   C
0  1.0   10.0  1
1  1.0   10.0  1
2  2.0    5.0  2
3  2.0    5.0  2
4  2.0    7.0  3
5  2.0    7.0  3
6  2.0    7.0  3

我怎样才能在熊猫数据框中做同样的事情？

标签： pythonpandasnan

解决方案

GroupBy.apply与前向和后向填充缺失值一起使用：

df[['A','B']] = df.groupby('C')['A','B'].apply(lambda x: x.ffill().bfill())
print (df)

     A     B  C
0  1.0  10.0  1
1  1.0  10.0  1
2  2.0   5.0  2
3  2.0   5.0  2
4  2.0   7.0  3
5  2.0   7.0  3
6  2.0   7.0  3

推荐阅读