3行ごとにリストに入れるプログラム

Question

前提・実現したいこと

下の「tekitou」のようなファイルを、一行ずつスペースで区切った形でさらに3行ごとにリストに入れるというプログラムを作りたいと思っています。

あ,い,う,え
か,き,く,け
こ,,
た,ち,,つ
な,に,ぬ,ね
は,ひ,ふ,へ
ま,み,む,

実現したい、リストをprintしたときの出力結果のイメージは以下の通りです。

['あ,い,う,え か,き,く,け こ,,', 'た,ち,,つ な,に,ぬ,ね は,ひ,ふ,へ', 'ま,み,む']

見辛いかもしれませんが、
「え」と「か」の間、「け」と「こ」の間、「つ」と「な」の間、「ね」と「は」の間がスペースで区切られており、また、
「あ」〜「こ,,」、「た」〜「へ」、「ま」〜「む」がそれぞれ別要素として扱われています。

発生している問題・エラーメッセージ

そして、実際に先の内容を実現するため、以下のようなプログラムを作成してみました。

# coding: UTF-8

#ライブラリをインポート
import codecs

#tekitouLogのログをスペース区切り3行ごとを配列に突っ込む方法を模索するプログラム

f_in  = codecs.open('tekitou.txt', 'r', 'utf-8')#実験用のログのファイルをutf-8で読み込み

lines = f_in.readlines() #読み込み
lines2 = [] #空のリストを用意

count = 0 #行数をカウントするための変数

threeline = "" #配列の要素を一時的に入れる変数

for line in lines:
    count += 1 #行数をカウント
    text = line.replace("\n","") #\n除去
    if count%3 != 0: #行数が3の倍数じゃなかった場合
        threeline += text #1行分をキープ
        threeline += ' ' #スペースを入れる
    elif count%3 ==0: #行数が3の倍数だった場合
        threeline += line #1行分をキープ
        lines2.append(threeline) #3行溜まったのでリストに追加
        threeline = "" #変数の中身をリセット

else:
    f_in.close()

print(lines2) #配列の中身を出力

しかし、これを実行すると、

['あ,い,う,え か,き,く,け こ,,', 'た,ち,,つ な,に,ぬ,ね は,ひ,ふ,へ']

のように、3行に入れず余った「ま,み,む」がリストに入りませんでした。
どのようにプログラムを組めば3行に入らず余ったものも含めた形でリストを作ることができるでしょうか。

Answer 1

現状のコードを最大限利用するなら、ループ後にこんなコードを埋めてやれば良いです。

if count % 3 != 0:
    lines2.append(threeline)

書いてみた

面白そうだったので。私ならこう書きます。

from itertools import zip_longest


with open('tekitou.txt') as fin:
    data = fin.read().split()
    
    
lines = [
    ' '.join(
        filter(lambda v: v is not None, block)
    )
    for block in zip_longest(*[iter(data)]*3)
]
print(*lines, sep='\n')

実行結果 Wandbox

あ,い,う,え か,き,く,け こ,,
た,ち,,つ な,に,ぬ,ね は,ひ,ふ,へ
ま,み,む,

Comments

[Terratale]

回答ありがとうございます。
そっか、たった2行追加するだけで理想通り動いたんですね！
itertoolsというライブラリも知りませんでした。知識が増えました、ありがとうございます！

Answer 2

リストの最後の要素を一時変数として使ってみました。

txt = """あ,い,う,え
か,き,く,け
こ,,
た,ち,,つ
な,に,ぬ,ね
は,ひ,ふ,へ
ま,み,む,
"""

result = []
for i, line in enumerate(txt.splitlines()):
    if i % 3 == 0:
        result.append(line)
    else:
        result[-1] += " " + line
    
print(result)  # => ['あ,い,う,え か,き,く,け こ,,', 'た,ち,,つ な,に,ぬ,ね は,ひ,ふ,へ', 'ま,み,む,']

Comments

[Terratale]

回答ありがとうございます。
enumerateという関数はじめて知りました。そういった方法もあるんですね！

Answer 3

def chunks(l, n):
    for i in range(0, len(l), n):
        yield l[i:i+n]


file_path = "data.txt"
with open(file_path) as f:
    lines = [line.rstrip("\n") for line in f]

x = [" ".join(line3) for line3 in chunks(lines, 3)]
print(x)

参考情報

• Break A List Into N-Sized Chunks

https://chrisalbon.com/python/data_wrangling/break_list_into_chunks_of_equal_size/

• Pythonのリストを決まったサイズで分割するときはmore-itertoolsのchunkedを使う

http://ariarijp.hatenablog.com/entry/2017/11/02/114539

• readlines（）を使うときに\ nを取り除く

https://code.i-harness.com/ja/q/e8713c