pythonのリストの要素取得について

Question

下記のような結果を取得するpythonのコードを書きたいです。
よろしくお願い致します。

配列：[4,4,2,4,5,3,1,3,4,1,2,2,3]
結果：[[6],[9,10,11]]

【テーマ】
または1,2・・・のインデックスを取得する

【内容】
配列に1がある場合
・2が続く場合は2が続くまでインデックスを返す
・2が続かない場合は終わる

よろしくお願い致します。

あまり良い書き方ではないですが、私のコードを記載します。
【追記】

result=[]
for i, x in enumerate(src):
    if x == 1:
        sample=[]
        sample.append(i)
        for j, y in enumerate(src[i+1:-1]):
            if y ==2:
                sample.append(i+j+1)
            else:
                break
        result.append(sample)

Answer 1

思い付きです。解説はしません。

import re


src = [4,4,2,4,5,3,1,3,4,1,2,2,3]
assert all(len(str(e)) == 1 for e in src)   # srcの全要素は1桁

dst = [
    [*range(m.start(), m.end())]
    for m in re.finditer(r'12*', ''.join(map(str, src)))
]

    
print(dst)

実行結果 Wandbox

[[6], [9, 10, 11]]

コメントを受けて

例えば、こんなふうに。

dst = []

i = 0
while i < len(src):
    if src[i] != 1:
        i += 1
        continue
        
    dst.append([i])
    i += 1
    
    while i < len(src) and src[i] == 2:
        dst[-1].append(i)
        i += 1

for文を使うなら二回に分けた方が無駄が少ないように思います。

dst = []

it = (
    i for i, e in enumerate(src)
    if e == 1
)
for index_1 in it:
    dst.append([index_1])
    
    idx = index_1 + 1
    for i, e in enumerate(src[idx:], start=idx):
        if e != 2:
            break
         
        dst[-1].append(i)

さらに追記

普通に読み辛いですし効率面も特段良くありませんが、個人的にはこういうのが好きです。
おまけです。

indice_1 = (
    i for i, e in enumerate(src)
    if e == 1
)
dst = [
    [idx, *map(operator.itemgetter(0), zip(
        itertools.count(idx+1),
        itertools.takewhile(
            lambda e: e == 2, src[idx+1:]
        )
    ))]
    for idx in indice_1
]

Comments

[ML_beginner]

for文を使わずに正規表現で記載している点が非常に参考になりました。
ご回答頂きありがとうございました。

[LouiS0616]

普通にwhile文を使った方が読み易く応用も効くと思いますよ。是非トライしてみて下さい。

[ML_beginner]

イテレーションを使ったやり方が分かりません。
理由としては、私の方針ではリストをfor文で回して1を見つけるというのがアプローチと考えました。
その際に1があった場合、次のイテレータの条件が書けません。
発見した1以降でfor文をもう一度回せば書けるとは思うのですが、あまり良いコードではないような気がしました。
お忙しい所恐縮ですが、方針だけでもご教示いただけないでしょうか。
【コード】
result=[]
for i, x in enumerate(src):
if x == 1:
sample=[]
sample.append(i)
for j, y in enumerate(src[i+1:-1]):
if y ==2:
sample.append(i+j+1)
else:
break
result.append(sample)

[LouiS0616]

しっかり考えておられるようなので（偉そうな言い方ですみません、しかし結果だけ欲しがる人がこのサイトには溢れかえっているのです）、回答に追記しました。

[ML_beginner]

ご丁寧にかつ迅速なご回答頂きまして誠にありがとうございます。
試したことなどを記載せず、簡潔な文章で質問を行なったことがかえって誤解を招いてしまいました。

またコードを頂き大変参考になりました。
・for文ではリストを二度作成することが無駄、ジェネレータ式で必要な部分だけ取り出す点
・while文ではi+1を行い、もう一度whileを使う点
・whileを使用すると同じ変数で二度ループできる点
・whileを使用することで一周分の処理になるため高速動作が期待できる点

今回コードレビューまでして頂き大変勉強になりました。
enumerateの引数指定の知識やジェネレータ使用のアイデア等他の人のやり方に触れる機会がないので、設計アイデアの欠如や知識不足を痛感しました。
改めて、誠にありがとうございました。