플래그 열에 모든 1이 있을 때 dataframe에서 그룹을 가져오고 크기가 2보다 큰 그룹을 유지하는 방법

df2=pd.DataFrame({'A':[1,20,40,45,56,1,20,40,45,56],'flag':[3,2,4,1,1,3,3,1,1,1]})
print(df2)
    A  flag
0   1     3
1  20     2
2  40     4
3  45     1
4  56     1
5   1     3
6  20     3
7  40     1
8  45     1
9  56     1

output
7  40     1
8  45     1
9  56     1

print(df2[(df2['flag']==1) & (df2['A']>2)])

s = df2["flag"].eq(1)
m = s.diff(1).ne(0).cumsum()
new_df = df2[s.groupby(m).transform(lambda x: x.sum()>2)]

    A  flag
7  40     1
8  45     1
9  56     1