Pythonを使って特定の文字を含む行とそれに続く２行の抽出を行いたい。

Question

前提・実現したいこと

テキストファイルから特定の文字列を含む行と、その行に続く下２行を抽出したい。

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

文字列を含む行は抽出できるのですが、下２行を抽出するコードが書けません。
教えてください。

該当のソースコード

Python

試したこと

ネットで調べたのですが、わかりませんでした。

今のところ作成したコードは以下のようになってます。

コード
# coding: utf-8

import re

f_in = open("テキスト","r")
f_out = open("テキスト2","w")

cnt = 0
for line in f_in:
 if cnt<=0:
   line = line.rstrip()
   m = re.search("特定の文字",line)
 if m:
    f_out.write(line)
    f_out.write("\n")
    cnt = 2
else:
    f_out.write(line)
    f_out.write("\n")
    cnt = cnt -1

f_in.close()
f_out.close()

Comments

[meg_]

> 文字列を含む行は抽出できるのですが、
そのコードを掲載してください。

[otn]

grepコマンドなら一発ですけど、プログラムを書くのが目的でしょうか？

Answer 1

ファイルを1行づつ読み込んでいき、条件に合うかどうかをチェック、
あったら、その行と続く2行を出すようにすればよろしい

Comments

[birdman]

回答ありがとうございます。
続く行を出す方法がわからなくて困っています。
どういうコードを使えばいいか教えてもらいたいです。

[y_waiwai]

ファイルから1行読み込みはできるんですよね？
なら、その読み込みと出力を2回繰り返せばいいだけです

[toast-uz]

テキストファイルから特定の文字列を含む行の抽出、という、できているところまでのコードを記載いただくと、アドバイスがしやすくなると思います。

[birdman]

遅くなって申し訳ありません。
今のところまでのコードを掲載しました。

[y_waiwai]

cnt = 0
for line in f_in:
if cnt<=0:
lines = line(1)
line = line.rstrip()
m = re.search("特定の文字",line)
if m:

f_out.write(line)
f_out.write(lines)
f_out.write("\n")
cnt = 2
else:
f_out.write(line)
f_out.write("\n")
cnt = cnt -1

こんなもんかなー

[birdman]

できました！ありがとうございます。

[birdman]

すいません、よく見たら他の場所も抽出してたみたいです。

# coding: utf-8

import re

f_in = open("テキスト","r")
f_out = open("テキスト","w")

cnt = 0
for line in f_in:
if cnt<=0:
line = line.rstrip()
m = re.search("特定の文字",line)
if m:
f_out.write(line)
f_out.write("\n")
cnt = 2
else:

f_out.write(line)
f_out.write("\n")
cnt = cnt -1

f_in.close()
f_out.close()

上記のコードで実行すると
特定の文字以外の場所も抽出してしまいます。

[y_waiwai]

cnt=2 のインデント位置がおかしいんでしょｗ
こんなベタ書きではインデント位置がわかんないので、
質問を編集し、<code>ボタンを押し、出てくる’’’の枠の中に、そのコードを貼り付けてください

[birdman]

いろいろ申し訳ないです。
編集してコードを加えました。
よろしくお願いします。

[y_waiwai]

if m:
のブロックのインデント位置が違います
直前の行とインデント位置を合わせてください

＃というか、どういうことをしているのかステップ実行させるなりして動作を理解しましょう

[birdman]

こんどこそできました。
本当にありがとうございます。
自分が何をしているか理解しながらできるように頑張ります。
ありがとうございました。