連続した値をカウントするリストで都度、最大値を抜き出したい。

Question

前提・実現したいこと

[0,1,0,0,1,2,0,1,2,3,4,5,6,0,0,1,0]のようなリストにおいて、
[1,2,6,1]のリストになるような出力をしたいのですが、上手くいきません・・・。
要件としては、1で加算していき、0でリセットするもので、リセットされる直前の最大値を取りたい形です。

試したこと

色々調べて、numpy配列にした上でscipy.signal.argrelmaxなども試したのですが、離散に対する関数だからか上手くいきませんでした。

補足情報（FW/ツールのバージョンなど）

python3

Answer 1

こちらも、0以外の要素は必ずカウントアップしているという仮定の下で。
（各パートの最大値は必ず 0 の直前の値となるので）

d = [0,1,0,0,1,2,0,1,2,3,4,5,6,0,0,1,0]

ret = [a for a,b in zip(d, d[1:] + [0]) if b == 0 and a != 0]
# [1, 2, 6, 1]

Comments

[sou10]

気付くの遅くなってすみません。
締め切ってしまっていたのにご回答頂き有難うございました！
1行で書けるんですね・・勉強になります。

Answer 2

0以外の要素は必ずカウントアップしているという仮定の下で。

import itertools

src = [0, 1, 0, 0, 1, 2, 0, 1, 2, 3, 4, 5, 6, 0, 0, 1, 0]
dst = []

for is_zero, it in itertools.groupby(src, key=lambda e: e == 0):
    if is_zero:
        continue
        
    elems = list(it)
    dst.append(len(elems))   # 二行まとめて、dst.append(sum(1 for _ in it)) でも可
    
print(dst)

実行結果 Wandbox

[1, 2, 6, 1]

Comments

[sou10]

ありがとうございます・・・！無事解決出来ました！！！！