条件指定した要素の連続値のカウントについて

Question

下記のようなarray型の要素があります。符号がプラスの値が最大何回連続したか、マイナスの値が最大何回連続したかを抽出したです。
下記の例だと、プラスは3回、マイナスが2回となるイメージです。

[0.1,0.5,0.03,-019,-0.05,0.1,0.5,-0.2]

ご教示頂けると幸いです。
どうぞよろしくお願い致します。

Comments

[mkgrei]

0が来た場合が悩みどころですが、普通はプラスにカウントするのでしょうか。

[NoPython_Nolife]

これは決め打ちしないといけないところですね。。今回はプラスカウントと考えております。

Answer 1

とりあえず『連続した回数』を取り出す方法。

from itertools import groupby

src = [0.1, 0.5, 0.03, -0.19, -0.05, 0.1, 0.5, -0.2]
dst = [sum(1 for e in it) for _, it in groupby(src, key=lambda x: x > 0)]
print(dst)

実行結果 Wandbox

[3, 2, 2, 1]

この中から正の値、負の値それぞれの連続する最大長を得たいの~~でしょうか？~~ですね。
それぞれmax(dst[0::2])およびmax(dst[1::2])で取得できます。
先頭の要素の正負に応じて使い分ければ良いです。

Comments

[NoPython_Nolife]

早速の回答ありがとうごいます！！！できました！ただ、ご説明いただきました、
「それぞれmax(dst[0::2])およびmax(dst[1::2])で取得できます。先頭の要素の正負に応じて使い分ければ良いです。」の部分が理解できておりません。。。。これはどういった意味なのでしょうか。。。大変恐縮ですが、この点補足して頂けると幸いです。

[LouiS0616]

dstは、『正の連続した数, 負の連続した数, 正の, 負の, ...』と並んでいるかもしれませんし、
『負の連続した数, 正の, 負の, ...』と並んでいるかもしれません。

dst[0::2]は『正の, 正の, 正の, ...』あるいは『負の, 負の, 負の...』となります。
どちらか判別するためにはdst[0]の正負を調べれば良い、ということです。

[NoPython_Nolife]

なるほど、そういう意味だったんですね！ありがとうごいます！！！すべて即答いただいた上に、大変分かりやすくご説明頂きほんとにありがとうございます。大変勉強になり、助かりました！

[hayataka2049]

groupbyを使うのなら、正負を1と-1に変換してsumすると、maxと-minで取り出せて好都合・・・というのはどうですか
result = [sum(items) for val, items in groupby([1 if x > 0 else -1 for x in lst])]

[LouiS0616]

ikapyさんの回答の良いところを吸い込んだ感じですね。大いにアリだと思います。

[LouiS0616]

別解として回答されてはいかがでしょうか。

[hayataka2049]

せっかくなのでそうしました

[hayataka2049]

せっかくなのでそうしました

Answer 2

正負を1と-1に変換しておくことで、maxと-minで取り出せるというアイデアです。

>>> lst = [0.1,0.5,0.03,-0.19,-0.05,0.1,0.5,-0.2]
>>> result = [sum(item) for val, item in groupby([1 if x > 0 else -1 for x in lst])]
>>> max(result)
3
>>> -min(result)
2

Comments

[ikapy]

よく考えたら、正だけとか負だけの場合はうまくいきませんね（私の場合もそうでした）。
ストレートにコードしてみました（.はスペースです）。なおデータがエンプティの場合の対処はしていません。

from itertools import groupby
test_list = [0.1, 0.5, 0.03, 0.19, 0.05, 0.1, 0.5, 0.2]
result = [sum(item) for val, item in groupby([1 if x > 0 else -1 for x in test_list])]
if len(result) == 1:
....if result[0] > 0:
........plus_ans = result[0]
........minus_ans = 0
....else:
........plus_ans = 0
........minus_ans = -result[0]
else:
....plus_ans = max(result)
....minus_ans = -min(result)
print(plus_ans, minus_ans)

[hayataka2049]

なるほど、確かに

[ikapy]

改良しました。

from itertools import groupby
test_list = [0.1, 0.5, 0.03, 0.19, 0.05, 0.1, 0.5, 0.2]
plus_result = [sum(item) for val, item in groupby([1 if x >= 0 else 0 for x in test_list])]
minus_result = [sum(item) for val, item in groupby([-1 if x < 0 else 0 for x in test_list])]
print(max(plus_result))
print(-min(minus_result))

[hayataka2049]

素直に一行にまとめてみたらこうなりました。numpy使うのは反則気味かもしれませんが
np.array([(lambda x:[x,0] if x > 0 else [0, -x])(sum(item)) for val, item in groupby([1 if x > 0 else -1 for x in lst])]).max(axis=0)

[NoPython_Nolife]

細かく修正までしていただき本当にありがとうございます！大変勉強になり、また助かりました！ありがとうございます！

[LouiS0616]

> よく考えたら、正だけとか負だけの場合はうまくいきませんね
resultに0を加えておくのはどうでしょう。

result = [sum(item) for val, item in groupby([1 if x > 0 else -1 for x in lst])] + [0]

[ikapy]

なるほど。感じ入りました。それだとエンプティリストの対応にもなりますしね。

[hayataka2049]

単純で良いですね。思いつかなかった

Answer 3

すでに回答が出てますが、せっかく考えたのでこんな方法もあるということで。
同一符号の場合は、カウントアップ（ダウン）するやりかたです。

test_list = [0.1, 0.5, 0.03, -0.19, -0.05, 0.1, 0.5, -0.2]
tmp_list = [0 for i in range(len(test_list))]
step = 0
for i, dat in enumerate(test_list):
    if dat > 0.0:
        if step < 0:
            step = 0
        tmp_list[i] += 1 + step
        step += 1
    else:
        if step > 0:
            step = 0
        tmp_list[i] += -1 + step
        step += -1
plus_counter = max(tmp_list)
minus_counter = -min(tmp_list)
print(plus_counter)
print(minus_counter)

Comments

[LouiS0616]

これは思いつかなかったです。良いですね。

[NoPython_Nolife]

このような方法もあるんですね！ありがとうございます！

Answer 4

似たような回答です。

このような
正の数 → a
負の数 → b
変換を元のリストに対して行い、それを空白なしで連結して文字列に整形します。
[0.1,0.5,0.03,-0.19,-0.05,0.1,0.5,-0.2]
↓
aaabbaab

この整形した文字列に対して、正規表現モジュールreのパターンマッチング（a+|b+）で連続した文字列を全て取り出します。
これをsortすることで連続したaの文字列のグループ、連続したbの文字列のグループに分割することができます（実際には１つのリスト）。
後は、リストのラストが負の数の最大連続数、それを除いた残りのリストの最大値が正の数の最大連続数になります。

import re

src = [0.1,0.5,0.03,-0.19,-0.05,0.1,0.5,-0.2]

src_t = ''.join(map(lambda x: 'ab'[x < 0], src))
groups = re.findall('a+|b+', src_t)

def seq_len_max(groups, s):
    return len(max(filter(lambda x: s in x, groups), key=len))

seq_plus = seq_len_max(groups, 'a')
seq_minus = seq_len_max(groups, 'b')

print(seq_plus, seq_minus)

Comments

[NoPython_Nolife]

こんなやり方もあったですね。非常に勉強になりました！ありがとうございます！

[arch_]

遊びで組んだので実行時間がアレですmm

Answer 5

array型がPythonのbuilt-inのarrayなのか、numpyのarrayなのかどっちでしょう。numpy.arrayを使ったほうがごちゃごちゃしたlambda式に頼らなくていいので読みやすなるような。

import numpy as np
src = [0.1,0.5,0.03,-0.19,-0.05,0.1,0.5,-0.2]
src = np.array(src)
is_positive  = (src > 0)
# array([ True,  True,  True, False, False,  True,  True, False])

あとは他の方の回答にある通り、groupbyを使うのがいいかなと思います。

from itertools import groupby
max_len = 0
for _is_positive, items in groupby(is_positive):
    if _is_positive:
        n = len(list(items))
        pos_max = n if n > pos_max else pos_max