Pythonテキストファイルから特定の行を取得

Question

前提・実現したいこと

pythonについての質問です。
以下のようなテキストファイルがあるとします。

テキスト1
テキスト2
正常終了
テキスト3
テキスト4
正常終了
テキスト5
テキスト6
エラー発生

テキストファイル全体を取得し、"エラー発生"という文字を発見した場合は"テキスト5"と"テキスト6"だけを抽出したいのですが、どういうやり方をすれば実現できるでしょうか？
よろしくお願いいたします。

Answer 1

１．readlines()でファイル全体をリストに読み込む
２．forでリストの添え字を変化させながら"エラー発生" in リスト要素で文字列を探し
３．見つかった添え字-1と-2のリスト要素が対象行

１行目や２行目に「エラー発生」があるケースがあるなら、添え字が範囲オーバーとならないような考慮も必要です。

grepコマンドが使えれば、grep -B2 エラー発生 ファイル名と一発ですが。

Comments

[qululu_123]

ご回答ありがとうございました。

Answer 2

Pandas使っていいなら

import pandas as pd

text = """テキスト1
テキスト2 
正常終了
テキスト3
テキスト4 
正常終了
テキスト5
テキスト6
エラー発生"""

df = pd.DataFrame(text.splitlines())

df["処理"] = df[0].where(df[0].isin(["正常終了", "エラー発生"])).fillna(method="bfill")

df[(df["処理"] == "エラー発生") & (df[0] != "エラー発生")][0].to_list()

Comments

[qululu_123]

ご回答ありがとうございました。

Answer 3

サンプルコード。

def main():
    data = []
    with open('text.txt', mode='r', encoding='utf-8') as f:
        for line in f:
            tmp = line.split('\n')[0]

            if tmp == '正常終了':
                data = []
            elif tmp == 'エラー発生':
                print('\n'.join(data))
            else:
                data.append(tmp)


if __name__ == '__main__':
    main()

Comments

[qululu_123]

ご回答ありがとうございました。

Answer 4

1. 変数１と変数２をクリア
2. １行読み込み、エラー発生、なら５に。
3. 変数２＝変数１、そして読み込んだ文字列を変数１に代入
4. ２にループ
5. 変数２と変数１を抽出
6. 1にループ

こんなもんでどうでしょう

＃いろいろ修正

Comments

[qululu_123]

ご回答ありがとうございました。
ご提示いただいたやり方で実装できました。