在Pandas数据帧中保留大小为1的块> = k

Question

我需要保留块＆gt; = k为'1'。所有其他“1”块应转换为零。例如， k = 2 ：

df= 
    a  b
0   1  1
1   1  1
2   0  0
3   1  0
4   0  0
5   1  0
6   0  0
7   1  0
8   0  0
9   1  1
10  1  1
11  1  1
12  0  0
13  0  0
14  1  0
15  0  0
16  1  1
17  1  1
18  0  0
19  1  0

其中列a是原始序列，列b是所需的。

Answer 1

z = df.a.eq(0)
g = z.cumsum().mask(z, -1)

k = 2
df['b'] = df.a.groupby(g).transform('size').ge(k).mask(z, 0)

    a  b
0   1  1
1   1  1
2   0  0
3   1  0
4   0  0
5   1  0
6   0  0
7   1  0
8   0  0
9   1  1
10  1  1
11  1  1
12  0  0
13  0  0
14  1  0
15  0  0
16  1  1
17  1  1
18  0  0
19  1  0