テキストファイルを部分的に出力

Question

前提・実現したいこと

テキストファイルの特定の部分のみ出力したいです。

報告書

〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇
〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇〇
×月第×週の平日予定

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
[月曜日の予定]買い物

[火曜日の予定]仕事

[水曜日の予定]仕事

[木曜日の予定]仕事

[金曜日の予定]映画鑑賞
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
土日は未定
のちの変更の可能性あり

このようなテキストファイルの、「~」の部分のみを出力するのに

with open("text.txt", "r") as f:
    flag = False
    for line in f:
        if line.startswith("~"):
            print(line)
            flag = not flag
        elif flag:
            print(line)
        else:
            pass

このようなコードを書きました。しかし、この「~」の中身の
「[×曜日の予定]」の後ろの部分のみを出力するのにはどうしたらよいかわかりません。
理想は

  買い物

  仕事

  仕事

  仕事

  映画鑑賞

このような形にできたらよいなと思っております。
お力添えいただければ幸いです。

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

python3.6

Answer 1

a.py

import re

with open("a.txt", "r") as f:
    m = re.match('(.*)(^~+$)(.*)(^~+$)(.*)', f.read(), flags=(re.DOTALL | re.MULTILINE))
    x = re.sub('[.*]', "", m.group(3))
    print(x)
    # tasks = filter(lambda x: len(x.strip()) > 0, x.splitlines())                                                          
    # print(list(tasks))  

実行例

コメントにしている部分を有効にすると、
['買い物', '仕事', '仕事', '仕事', '映画鑑賞']
のように list を生成し、それを画面に出力できます。

参考情報

• Python Tips：正規表現で複数行を扱いたい

https://www.lifewithpython.com/2017/12/python-regular-expression-multiple-lines.html

Comments

[yamidora]

簡潔かつ正確な回答ありがとうございます！
list生成の処理まで書いていただき、とても助かります！

Answer 2

正規表現で二段階に分けて片付けた方が、すっきりしたコードになると思います。
フラグを利用したコードは難解になりがちです。

import re


with open('data.txt') as fin:
    data = fin.read()
    
m = re.search(r'(~{10,}).+(\1)', data, flags=re.DOTALL)
assert m

data = m.group(0)
# print(data)

for item in re.finditer(r'^[.曜日の予定](.+)$', data, flags=re.MULTILINE):
    print(item.group(1))

実行結果 Wandbox

買い物
仕事
仕事
仕事
映画鑑賞

Comments

[yamidora]

ありがとうございます！
やったことのない処理なのですこし難しく感じますが、チャレンジしてみます！

Answer 3

あなたのスクリプトに多少手を加えました。

import re
with open("text.txt", "r") as f:
    flag = False
    for line in f:
        if line.startswith("~"):
            print(line)
            flag = not flag
        elif flag:
            mt = re.search(r'(^\s*[(月|火|水|木|金|土|日)曜日の予定])(.*)',line)
            if mt != None:
                print(mt.group(3))
        else:
            pass

Comments

[yamidora]

ありがとうございます！
このようなやり方は思い浮かばなかったので勉強になりました！

Answer 4

出力例が、以下のようになっていますが、これは、[X曜日の予定]後の他に、その次の行も出力したいということですか。
買い物

仕事

仕事

仕事

映画鑑賞

もし、[X曜日の予定]の後だけでよいなら、
買い物
仕事
仕事
仕事
映画鑑賞
のようになりますが。

Comments

[yamidora]

[X曜日の予定]の後だけが出力させたいものです
表現不足で申し訳ありません。

Answer 5

フォーマットが決まっているようなので、t_obaraさんの回答にあるようにsplitを使用して']'文字のところで分割するのが簡単だと思います。もしくは正規表現を使用して[…]の部分とそれ以降を分けてもいいかもしれません。
今回のケースで該当するのかは分かりませんが、[月曜日の予定]買い物[メモ: …] のように後半にも[]が登場する場合は注意が必要です。

import re

with open("text.txt", "r", encoding='utf-8') as f:
    flag = False
    for line in f:
        if line.startswith("~"):
            print(line)
            flag = not flag
        elif flag:
            # split()で分割
            if ']' in line:
                tmp = line.split(']', maxsplit=1)
                print(tmp[-1])

            # 正規表現を使う場合 (import reが必要)
            r = re.match(r'([.*?])(.*)', line)
            if r:
                print(r.group(2))
        else:
            pass

Comments

[yamidora]

二通りもやりかたを教えて頂きありがとうございます！
思った通りの処理ができそうです！

Answer 6

lineに空白で区切られて入っていると思うので、splitを利用して分割すると良いのではないでしょうか。

Comments

[yamidora]

お早い回答ありがとうございます。
大変申し訳ありません。テキストファイルの内容を確認したところ、[]の後ろに空白が入って
おりませんでした。質問文を入力する際に元のテキストが手元になく、テキストを貼り付けず直接質問文に入力したため、本来のもとの違う内容になってしまいました。テキストの内容を直接貼り付けたものに修正致しました。