python - 根据另一行的值更新数据框中的行值？

Question

我有一个包含两列的数据框:a 和 b

df

         a         b
0       john      123
1       john
2       mark     
3       mark      456
4       marcus    789

我想根据 a 列更新 b 列的值。

         a         b
0       john      123
1       john      123
2       mark      456
3       mark      456
4       marcus    789

如果 john 在 b 中有值 123。剩余的 john 也必须具有相同的值。

Answer 1

假设您的数据框是:

df = pd.DataFrame({'a': ['john', 'john', 'mark', 'mark', 'marcus'], 'b': [123, '', '', 456, 789]})

您可以 df.groupby 列 a 上的数据帧，然后在返回第一个非空值的分组数据帧的列 b 上应用 transform在分组列 b 中。

使用:

df['b'] = (
    df.groupby('a')['b']
    .transform(lambda s: s[s.ne('')].iloc[0] if s.ne('').any() else s)
)

结果:

# print(df)

        a    b
0    john  123
1    john  123
2    mark  456
3    mark  456
4  marcus  789