Pandas - 用另一列的字符串填充 NaN

Question

我有 2 列（“姓氏”和“邮政编码”）。 dataframe 已过滤为仅包含重复的姓氏：

Surname | PostCode
Adams   | NaN
Adams   | NaN
Bryan   | NX203
Bryan   | NaN
Cormack | NaN
Cormack | NaN
Cormack | NZ233
Dylan   | NaN
Dylan   | NaN
Dylan   | NaN

其中一些根本没有邮政编码。 然而，对于那些有的人，我想用现有的任何东西来填补缺失的部分。 例如，包含“Bryan”的第二行应填充 NX203（就像上面的行一样）。 类似地，其他两个 Cormack 实例应填充 NZ233。

我不知道从哪里开始。 我假设它必须是应用于每一行的 python function 但不确定如何开始/做什么。

Answer 1

让我们试试groupby().transform() ：

df['PostCode'] = df.groupby('Surname').PostCode.transform('first')

Output：

   Surname PostCode
0    Adams      NaN
1    Adams      NaN
2    Bryan    NX203
3    Bryan    NX203
4  Cormack    NZ233
5  Cormack    NZ233
6  Cormack    NZ233
7    Dylan      NaN
8    Dylan      NaN
9    Dylan      NaN

Answer 2

另一种方式， groupby(), ffill再bfill

df['PostCode'] =df.groupby('Surname').PostCode.apply(lambda x:x.ffill().bfill())



 Surname PostCode
0    Adams      NaN
1    Adams      NaN
2    Bryan    NX203
3    Bryan    NX203
4  Cormack    NZ233
5  Cormack    NZ233
6  Cormack    NZ233
7    Dylan      NaN
8    Dylan      NaN
9    Dylan      NaN