Pythonを使ったデータフレーム操作

Question

###データフレーム操作(pandas)
連続する整数の場合、足し算をする。

###データフレーム例
データフレームdfにはカラムAとBがあり、Bの内1が4回連続した場合のみを抽出したいのですが
どのように抽出すればいいかわかりません。

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

###試したコード
df[df['B'] >=1]
で実行するとカラムBの1は抽出できるがやりたいことは
カラムAの4~7の1が連続した部分のみを取り出したいです。

何かいい方法があるかご教示いただけますか。
よろしくお願いします。

追加となりますが、
仮にデータフレームを毎回出力するたびに
カラムBの値がランダムに0,1を生成されるとした場合、カラムAを利用してなんとか連続する場合のみを取得したいです。

A   B
2   1
4   1
5   1
6   1
7   1
9   1

Answer 1

pandas.DataFrame.rollingを使うと、連続したデータの集計ができます。
これを利用して、対象の行から4つ前までの値の合計値を出し、これが4になったものが該当します。

import pandas as pd

df = pd.DataFrame({'A': range(11), 'B': [0, 0, 1, 0, 1, 1, 1, 1, 0, 1, 0]})
consecutive = 4
ix = df.loc[df['B'].rolling(consecutive).sum() == consecutive].index
for x in ix:
    print(df.loc[x - consecutive + 1:x])

上記のコードは1が5つ以上続いてしまうと、データが重なってしまうことにご注意ください。

Comments

[marchpython]

ありがとうございます。解決できました。

Answer 2

すいません。回答を間違えました

Comments

[marchpython]

ご回答ありがとうございます。
言葉足らずの部分があり、ご不敏をおかけしてすいません。