python - 根据列的值在 DataFrame 中填充 NaN

Question

我有类似于以下简化示例的数据:

Col1    Col2    Col3
a       A       10.1
b       A       NaN
d       B       NaN
e       B       12.3    
f       B       NaN
g       C       14.1
h       C       NaN
i       C       NaN

...对于数千行。我需要根据 Col2 中的值填充 na，使用类似于 ffill 方法的东西。我正在寻找的结果是这样的:

Col1    Col2    Col3
a       A       10.1
b       A       10.1
d       B       NaN
e       B       12.3    
f       B       12.3
g       C       14.1
h       C       14.1
i       C       14.1

但是，此方法会忽略 Col2 中的值。有什么想法吗？

Answer 1

如果我理解正确，那么您可以在“Col2”上分组，然后在“Col3”上调用转换并调用 ffill:

In [35]:

df['Col3'] = df.groupby('Col2')['Col3'].transform(lambda x: x.ffill())
df
Out[35]:
  Col1 Col2  Col3
0    a    A  10.1
1    b    A  10.1
2    d    B   NaN
3    e    B  12.3
4    f    B  12.3
5    g    C  14.1
6    h    C  14.1
7    i    C  14.1