python - 如何用另一个匹配行的结果填充 NaN？

Question

我有一个格式为的数据框

USER | ADDRESS | PURCHASE
1    | add 1   |  A
1    | NaN     |  B
2    | NaN     |  C
2    | add 2   |  D
3    | NaN     |  E

对于用户 1，我想用 add 1 填充 NaN 行中的地址，并且我想对用户 2 的 NaN 行执行相同的操作。我不想分组，我想保留 4 行格式。只是想填写NAN地址。对于没有匹配项的用户 3，我想将其保留为 NaN。

我尝试过创建一个 groupby 用户，但这最终会填充所有内容并将所有内容放在一起，这不是我想要的

没有代码

我只想相应地填写每个用户的地址

Answer 1

我不确定分组有什么问题，但是当你想“按组”填充缺失值时，你需要使用groupby

df['ADDRESS'] = df.groupby('USER').ADDRESS.apply(lambda x: x.ffill().bfill())

   USER ADDRESS PURCHASE
0   1   add 1   A
1   1   add 1   B
2   2   add 2   C
3   2   add 2   D
4   3   NaN     E