0,1配列内で1が連続でx個登場したタイミングで、ループを停止するようなプログラム

Question

#実現したいこと
0と1からなる配列(ここではわかりやすいように1*100行列として)に関して、
この行列内で、指定の個数分以上1が連続した時、forループを停止し、
その1が連続する並びの先頭の1が行列の何番目かを表示するような
アルゴリズムを作成したいです。

#例
1が3個以上連続したとき、その連続する1の先頭を表示
[0 0 0 0 1 0 0 0 1 1 1 1 1 0 0 0 0]

上記の場合　答えは　8

#自力で作成できたもの

A = [1 0 0 1 1 0 1]
a = 0
for num in A:
    a = a + 1
    if num == 1:
        print('point: {:d}'.format(a))
        break

上記のように配列の先頭から最初に1が登場したタイミングの検出はできましたが、
初心者のため、これを拡張して作っていこうかと考えたのですが、
手が止まってしまいました。

#環境
python3.7

Comments

[stdio]

「アルゴリズムを作成したいです。」と言われましても、試したソース等がないならアドバイス出来ません。

[pypypy.bom]

追加させていただきました。

[8524ba23]

例の答えは８では？

[pypypy.bom]

失礼いたしました。訂正いたします。

Answer 1

文字列にしてre.serach()で先頭を探して見つけます。

import re

A = [0,0,0,0,1,0,0,0,1,1,1,1,1,0,0,0,0]
N = 3 # 1の連続数

s = ''.join(map(str,A)) # リスト→文字列
m = re.search('1'*N,s)
if m:
    print(m.start()) # 8

Answer 2

itertools.groupby を使うと直感的に書けます。

from itertools import groupby

lst = [0, 0, 0, 0, 1, 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0]
N = 3  # 「N 個以上連続していたら」という条件の値

i = 0
for k, g in groupby(lst):
    g = list(g)  # list に変換

    if g[0] == 1 and len(g) >= N:
        # 1がN 個連続して並んでいる場合、最初のインデックスを出力して終了する。
        print(i)
        break

    i += len(g)

8

itertools.groupby の使い方

from itertools import groupby
lst = [0, 0, 0, 0, 1, 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0]

for k, g in groupby(lst):
    print(k, list(g))
# 0 [0, 0, 0, 0]
# 1 [1]
# 0 [0, 0, 0]
# 1 [1, 1, 1, 1, 1]
# 0 [0, 0, 0, 0]

自前で書く場合

lst = [0, 0, 0, 0, 1, 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0]
N = 3

curr_value = lst[0]
cnt = 0
for i, value in enumerate(lst):
    if value == curr_value:
        # 前回と同じ値ならカウント +1
        cnt += 1
    else:
        # 前回と異なる値ならカウントを0にする。
        cnt = 0
        curr_value = value
    
    if cnt == N and value == 1:
        # N個連続 かつ 値が1の場合ループ終了
        print(i - N)
        break

Comments

[pypypy.bom]

ありがとうございます!!
itertoos、初めて知りました！
利用させていただきます！

Answer 3

素直にforで書くとこんな感じだと思います。それほど複雑ではないものの、少し条件がややこしいかもしれません。

A = [0, 0, 0, 0, 1, 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0]
n = 3

cnt = 0
result = None
for i, num in enumerate(A):
    if num == 0 and cnt != 0:  # nまで数える前に0が出たらcntを0に
        cnt = 0
    if num == 1: # 1なら1加算
        cnt += 1
    if cnt >= n:  # n以上1が続いたら位置を格納してbreak
        result = i - n + 1
        break
print(result)

横着。

>>> l = [0, 0, 0, 0, 1, 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0]
>>> s = "".join(map(str, l))
>>> s.find("111")
8