python - Pandas 根据上一个可用值和下一个可用值填充 NaN

Question

我的 csv 数据如下所示:

     A     B
0   x aa   
1   z aa   
2          
3   
4   x aa   
5   z bb
6   x bb
7          
8   z cc   

如果在 A 列中，我想用 A 中的值填充 B 列中的空单元格

last_available_value_before_the_NaNs_in_A.split()[-1] == next_available_value_after_the_NaNs_in_A.split()[-1]

想要的结果是:

     A     B
0   x aa   aa 
1   z aa   aa
2          aa
3          aa
4   x aa   aa
5   z bb   bb
6   x bb   bb
7          
8   z cc   cc

data.loc(7,'B') 将为 NaN 因为 data.loc(6,'A').split()[-1] ==
data.loc(8,'A').split()[-1] 为假。

data.loc(5,'B') 是 'bb' 因为 data.loc(5,'A').split()[-1] == 'bb' '

感谢您的帮助!

Answer 1

您可以比较使用 ffill 的版本和使用 bfill 的版本:

f = df.A.fillna(method='ffill').str.split().str[-1]
b = df.A.fillna(method='bfill').str.split().str[-1]
df.B.where(f != b, f, inplace=True)