python 条件一致で文字をつなげていく

Question

list hoge 1 2
list hoge2 3 4
list hoge3 5 6
list hoge4 7 8
list hoge5 1 2
list hoge6 3 4
list hoge7 1 2
list hoge8 3 4
list hoge9 5 6

data = []
with open("data", "r") as f:
    for line in f:
        if line.startswith("list"):
            data.append(line.split())
            size = len(data)
            print(size)
            for i in range(size - 1):
                first, next = data[i], data[i+1]
                ("{}-{}".format(first[1], next[1]))
                if int(next[2]) > int(first[3]):
                    data = []
                    print(c[1], n[1])

#理想的な出力結果
#hoge-hoge2-hoge3-hoge4
#hoge5-hoge6
#hoge7-hoge8-hoge9

hogeに与えた２つの数値のうち後者（hoge x y）のyが次の行のhogeのxの数値よりも
大きい場合にhogeをつなげていきたいです。
大きい場合にどんどんつなげていく方法がわからずに困っています。

1行ずつ読み込むよりも一旦ずべてリストに入れてから処理する方がいいのかとも考えております。

できればpandasを使わずに1行ずつの処理で求めてみたいです。

アドバイスお願い致します。

Answer 1

常にhogeの部分を書き出して、条件次第でハイフンか改行文字を付け加えるかを選択していけばいいのではないでしょうか。あんまりPythonらしくない書き方かもしれませんけども。

from os import linesep
from sys import stdout
from io import StringIO
import csv

s = """list hoge 1 2
list hoge2 3 4
list hoge3 5 6
list hoge4 7 8
list hoge5 1 2
list hoge6 3 4
list hoge7 1 2
list hoge8 3 4
list hoge9 5 6"""

# with open(fname, "r") as f: # ファイルから読み込む場合はこちらを使う
with StringIO(s) as f:
    reader = csv.reader(f, delimiter=" ")

    # 一行目は比較対象がないのでforの前に処理しておく。
    (l, hoge, *prev) = next(reader)
    stdout.write(hoge)

    # read remaining rows
    for (l, hoge, *row) in reader:
        if int(row[0]) > int(prev[1]):  # 連結の条件
            stdout.write("-")           # 連結文字
        else:
            stdout.write(linesep)       # 改行文字
        stdout.write(f"{hoge}")
        prev = row[:]
    stdout.write(linesep)               # 必要に応じて
    stdout.flush()                      # 念のためflushしておく

StringIOの部分はデバッグ用です。

Comments

[hayataka2049]

stdout.writeしなくてもprint(hoge, end="")で十分では？

[tachikoma]

それでもいいですね。直接バッファに書き込むイメージでコードを書いたので、stdoutむき出しにしてました。

[22Go]

ありがとうございます！

Answer 2

４個目のデータを比較して、前の行より大きかったら２個目のデータchunkリストに格納、
小さかったらchunkリストを結合して dataリストに格納・・・・という処理で実現できるかと思います。

data = []
chunk = []
prev = -1
with open("data", "r") as f:
    for line in f:
        l = line.strip().split()
        if (len(l) != 4) or (l[0] != 'list'):
            continue
        if prev < int(l[3]):
            chunk.append(l[1])
        else:
            data.append('-'.join(chunk))
            chunk = [l[1]]
        prev = int(l[3])
    data.append('-'.join(chunk))

print('\n'.join(data))
#hoge-hoge2-hoge3-hoge4
#hoge5-hoge6
#hoge7-hoge8-hoge9

Answer 3

いったんリストに格納しました。比較のためにzipを使いました。文字列結合結果を作るのにfoldlを使いたかったのですが面倒なのでやめました。

data = []
with open("data", "r") as f:
    rows = list(map(lambda x: (x[1],int(x[2]),int(x[3])),[(l.split()) for l in f if l.startswith("list")]))
data.append(rows[0][0])
for (_,_,e),(h,s,_) in zip(rows[:-1],rows[1:]): # for x,y in zip(rows[:-1],rows[1:]):
    data.append(("-" if e < s else "\n")) # data.append(("-" if x[2] < y[1] else "\n"))
    data.append(h) # data.append(y[0])
else:
    print(''.join(data))

上記のforの変数名を変更しました。x,y --> (,,e),(h,s,_)

別解
functools.reduceを使うと短くなりました。pythonらしくないかもしれません。

import functools    
with open("data", "r") as f:
    rows = list(map(lambda x: (x[1],int(x[2]),int(x[3])),[(l.split()) for l in f if l.startswith("list")]))
(h,_,_)=functools.reduce(lambda x,y: (x[0] + ("-" if x[2] < y[1] else "\n") + y[0] ,y[1] ,y[2]), rows)
print(h)

Comments

[22Go]

非常に参考になりました。ありがとうございました。

Answer 4

import io

s = """list hoge 1 2
list hoge2 3 4
list hoge3 5 6
list hoge4 7 8
list hoge5 1 2
list hoge6 3 4
list hoge7 1 2
list hoge8 3 4
list hoge9 5 6
"""

before_y = 0
result = [[]]
for line in io.StringIO(s):
    _, hogestr, x, y = line.split()
    x, y = int(x), int(y)
    if x > before_y:
        result[-1].append(hogestr)
    else:
        result[-1] = "-".join(result[-1])
        result.append([hogestr])
    before_y = y
result[-1] = "-".join(result[-1])
print(result)  # => ['hoge-hoge2-hoge3-hoge4', 'hoge5-hoge6', 'hoge7-hoge8-hoge9']

Comments

[22Go]

とてもシンプルでわかりやすいコードでした。
ありがとうございました。