python 複数条件分岐

Question

list hoge1 1 2 +
list hoge2 3 4 + 
list hoge3 15 16 -
list hoge4 17 18 -
list hoge5 19 20 +
list hoge6 31 32 +
list hoge7 33 34 +
list hege8 45 46 -

data = []
data_plus = []
data_minus = []
chunk = []
chunk_plus = []
chunk_minus = []
prev = 0
with open("data","r")as f:
    for line in f:
        new_line = line.strip().split()

        if not line:
            continue

         if new_line[4] != "+" and int(new_line[2]) - prev < 10:
            chunk_plus.append(new_line[1])

        else:
            data_plus.append('-'.join(chunk_plus))
            chunk_plus = [l[1]]

        if new_line[4] != "-" and int(new_line[2]) - prev < 10:
            chunk_minus.append(new_line[1])

        else:
            data_minus.append('-'.join(chunk_minus))
            chunk_minus = [new_line[1]]

        prev = int(l[3])

    data_plus.append('-'.join(chunk_minus))
    data_minus.append('-'.join(chunk_plus))


print('\n'.join(data_plus))
print('\n'.join(data_minus))

#実行結果
hoge1
hoge2
hoge3-hoge4
hoge5
hoge6
hoge7
hege8
hoge1-hoge2
hoge3
hoge4-hoge5
hoge6-hoge7
hege8

hoge1-hoge2
hoge3-hoge4
hoge5
hoge6-hoge7
hoge8

という結果を出したいです。
数字の隣が+の時　かつ　数字の差が10以下であればhogeをつなげる
数字の隣が-の時　かつ　数字の差が10以下であればhogeをつなげる
数字の差が10以下であっても+と-が一致していなければhogeはつなげない様にしたいです

アドバイスいただきたです、宜しくお願い致します。

Comments

[coco_bauer]

書いたプログラムを質問に追加してください。　そのプログラムを実行した結果も追加してください。

Answer 1

data_plusとdata_minusで一度値を保存していますが、別々に保存するとそれぞれ2度表示されてしまうので、この二つの変数は消しましょう。

提示されている条件文をまとめると、hogeをつなげる場合は「前の行の符号と一致している かつ 数字の差が10以下」のみです。
よって、以下の3つの方策でこれは実現できます。

1. 前の行の符号と一致している かつ 数字の差が10以下ならハイフンを表示
2. そうでないなら改行を表示
3. hogeを表示

書き直しました。

with open("data", "r") as f:
    tmp, hoge, prev, tmp, psign = f.readline().split()
    prev = int(prev)
    print(hoge, end="")

    for line in f:
        values = line.strip().split()
        if int(values[2]) - prev < 10 and values[4] == psign:
            print("-", end="")
        else:
            print()
        print(values[1], end="")

        prev = int(values[3])
        psign = values[4]
    print()

withの後の3行は初期化。
for文の中でprevとpsign(previous-sign)を保存しておき、次の行の数字の差と符号を比較しています。

Comments

[tachikoma]

サンプルを書くのであればもう少し実行できる形にしたほうがいいです。

[22Go]

ありがとうございます。日本語にして考えると理解しやすいことがわかり、大変助かりました。

[22Go]

すみません、追加で質問させてください。
hogeをつなげた個数の結果を表示したいのですが可能ですか？
最終的にヒストグラムを作成したいと考えています。

[kokorin]

可能です。
hogeをカウントする変数を作り、hogeを繋げる場合はカウントする変数を+1し、行が切り替わるタイミングで表示すればいけます。
hogeをつなげた個数の結果を保持する場合は行が切り替わるタイミングでlistにappendして下さい。
別回答でサンプルコードを書いておきます。

[22Go]

シンプルな方法で非常に理解しやすいです。ありがとうございます！

Answer 2

hogeをつなげた個数の結果を表示するサンプルコード

with open("data", "r") as f:
    tmp, hoge, prev, tmp, psign = f.readline().split()
    prev = int(prev)
    count = 1                    # ここでカウント変数を定義
    print(hoge, end="")

    for line in f:
        values = line.strip().split()
        if int(values[2]) - prev < 10 and values[4] == psign:
            print("-", end="")
            count += 1           # hogeをつなげるときに変数を+1する
        else:
            print("", count)     # 行が切り替わるタイミングで表示(またはlist.append)
            count = 1            # カウント変数をリセットする
        print(values[1], end="")

        prev = int(values[3])
        psign = values[4]
    print("", count)