リストから対象の塊のみを抽出する。

Question

お世話になります。
リストから対象の塊を抽出して再度リストで保持したいです。
以下のようなイメージです。

list1 = [0,0,1,1,1,0,0,1,1,0]
list2 = func(list1,1)
# list2 = [[1,1,1],[1,1]]

for文を使って工夫すれば出来るのですが、もっとスマートな方法（ライブラリ等）あれば教えて頂ければと存じます。
よろしくお願いいたします。

Answer 1

標準のitertools.groupbyなどでどうでしょうか。

from itertools import groupby

list1 = [0,0,1,1,1,0,0,1,1,0]
print([list(g) for k, g in groupby(list1) if k == 1]) # 簡単なので解説なしです

itertools --- 効率的なループ実行のためのイテレータ生成関数 — Python 3.8.0 ドキュメント

Answer 2

itertools.groupby が使えます。

import itertools


src = [0, 0, 1, 1, 1, 0, 0, 1, 1, 0]
dst = [
    list(it)
    for tf, it in itertools.groupby(src, key=lambda e: e == 1) if tf
]

print(dst)

実行結果 Wandbox

[[1, 1, 1], [1, 1]]

Comments

[luluca]

お二人ともありがとうございます！
まさに望んでいたやつでした！