pandas DataFrame设置了非连续的部分

Question

我有一个如下所示的DataFrame，并希望在列A中的1（其中n = 2以下）之后，对于n行，B为1

index  A   B
0      0   0
1      1   0
2      0   1
3      0   1
4      1   0
5      0   1
6      0   1
7      0   0
8      1   0
9      0   1

我想我可以使用类似于这个例子的.ix来做，但不知道如何。如果可能的话，我想在一个熊猫式的选择命令中做到这一点。 （理想情况下不使用rolling_apply。） Modifying a subset of rows in a pandas dataframe

编辑：应用程序是A列中的1被“忽略”，如果它落在前一行的n行中。根据注释，对于n = 2然后，这些示例：

A = [1,0,1,0,1]，B应为= [0,1,1,0,0] A = [1,1,0,0]，B应为[0,1,1,0]