python - python / Pandas : how to forward fill from specific value within a group

Question

我在 pandas 数据框中的数据如下所示:

      dummy group
0       0    A1
1       0    A1
2       0    A1
3       1    A1
4       0    A1
5       0    A1
6       0    B2
7       0    B1
8       0    B2
9       0    B2
10      0    B2
11      0    B2

我正在尝试用其他值填充第一个 1 中 A1 的其余值。使用 ffill 去除 NaN 非常简单，但我真的可以在此条件填充上使用一些帮助。谢谢

编辑:

结果应该是这样的:

      dummy group
0       0    A1
1       0    A1
2       0    A1
3       1    A1
4       1    A1
5       1    A1
6       0    B2
7       0    B1
8       0    B2
9       0    B2
10      0    B2
11      0    B2

Answer 1

如果我没看错，你想要从第一个开始，并且只有 1 和 0 可能出现在 dummy 中，那么你可以使用 numpy cumsum:

>>> df['dummy'] = df.groupby('group')['dummy'].transform(np.cumsum)
>>> df.ix[df['dummy']!=0, 'dummy'] = 1
>>> df
    dummy group
0       0    A1
1       0    A1
2       0    A1
3       1    A1
4       1    A1
5       1    A1
6       0    B2
7       0    B1
8       0    B2
9       0    B2
10      0    B2
11      0    B2

[12 rows x 2 columns]