重复 B 列中的值，直到 python 中 A 列发生变化

我是 python 新手，有一个疑问。我需要在 B 列中重复该值，直到 A 列发生更改。

这是示例数据：

A   B
18  1
18  0
18  0
24  2
24  0
24  0
24  0
10  3
10  0
10  0

我想要怎样的输出

Column A  Column B
18         1
18         1
18         1
18         1
24         2
24         2
10         3
10         3
10         3
10         3

请帮助我解决这个问题。谢谢

您可以使用transform by first如果需要重复每组的第一个值：

df['Column B'] = df.groupby('Column A')['Column B'].transform('first')
print (df)
   Column A  Column B
0        18         1
1        18         1
2        18         1
3        18         1
4        24         2
5        24         2
6        10         3
7        10         3
8        10         3
9        10         3

另一种解决方案不依赖于Column A - replace 0值由NaN，使用前向填充ffill最后投射到int:

df['Column B'] = df['Column B'].replace(0,np.nan).ffill().astype(int)
print (df)
   Column A  Column B
0        18         1
1        18         1
2        18         1
3        18         1
4        24         2
5        24         2
6        10         3
7        10         3
8        10         3
9        10         3