python 指定文字を含む行が2回以上続いた場合に、行を削除

Question

前提

python初心者です。
自分の勉強不足なのは理解しているので、自分で調べろ勉強しろなどといった答えは控えて頂けると幸いです。

今回はpythonで読み込んだテキストファイルの指定した行を削除するようなプログラムを作成しています。

実現したいこと

実際に指定した行に含まれる文字を指定することで、削除するようなプログラムは作成することができました。
ただ自分のやりたいこととしては、その指定した文字を含む行が2回以上続いた場合に、2行目以降を削除するといったプログラムです。

例えば、
・見積書
・注文書
・請求書
＋ ・売上管理表
・行程管理表
ー ・売上管理表
＋ ・P/S
＋ ・B/S
ー ・利益
ー ・合計
＋ ・その他
テキストファイルの中がこのようになっている時、＋が指定文字とするなら2回以上続いた・B/Sの行を削除したいというような感じです。

発生している問題

自分の勉強不足であることは非常にわかっておりますが、「2回以上続く場合」といったような条件を自分のソースコードにどのように書き込めばいいのか思い当たらず質問させていただいております。

該当のソースコード

inputFile = 'sample.txt'
outputFile = 'sample_2.txt'

containWord = []    
NGWord = ['+']    

for line in open(inputFile):
    # 行中にNGWordを含むか判別
    for i in NGWord:
        if i in line:
            break
    else:
        # 行中にcontainWordを含むか判別
        for i in containWord:
            if i not in line:
                break  
        else:
            # NGWordを含まないかつcontainWordを含む場合、
            # 書き込みファイルの末尾にその行を追加
            with open(outputFile, mode = 'a') as f:
                f.write(line)

Comments

[otn]

> その指定した文字を含む行
というのと、プログラム中の containWord 、 NGWord の関係が不明です。

Answer 1

import io
from itertools import groupby

lines = '''
・見積書
・注文書
・請求書
＋ ・売上管理表
・行程管理表
ー ・売上管理表
＋ ・P/S
＋ ・B/S
ー ・利益
ー ・合計
＋ ・その他
'''.strip()
inputText = io.StringIO(lines)
outputText = io.StringIO()

containWords = ['＋']
for i, v in groupby(inputText, key=lambda l: any(w in l for w in containWords)):
    outputText.write(next(v) if i else ''.join(v))

print(outputText.getvalue())
outputText.close()

-が2回以上続いた際に最後の-の行以外は削除する

containWords = ['ー']
for i, v in groupby(inputText, key=lambda l: any(w in l for w in containWords)):
    outputText.write([*v][-1] if i else ''.join(v))

テキストファイルを読み込みたい場合はどうすればいいでしょうか？

from itertools import groupby

inputFile = 'sample.txt'
outputFile = 'sample_2.txt'

containWord = ['＋']
with open(inputFile, 'r') as infile, open(outputFile, 'w') as outfile:
    for i, v in groupby(infile, key=lambda l: any(w in l for w in containWord)):
        outfile.write(next(v) if i else ''.join(v))

# containWord = ['ー']
# with open(inputFile, 'r') as infile, open(outputFile, 'w') as outfile:
#     for i, v in groupby(infile, key=lambda l: any(w in l for w in containWord)):
#         outfile.write([*v][-1] if i else ''.join(v))

Comments

[Alex]

参考のソースコード誠にありがとうございます。
可能であれば、-が2回以上続いた際に最後の-の行以外は削除するといったソースコードも教えて頂ければと思います。
不躾ではありますが、よろしくお願い致します。

[Alex]

重ね重ねすみません。
こちらテキストファイルを読み込みたい場合はどうすればいいでしょうか？
自分で挑戦してもうまくできなくて、教えて頂けると助かります。

Answer 2

以下コードのように連続した回数を保持しておき、この回数によって出力判断すればよいです。
ループ中にて連続してなければ回数を０にリセットします。
なお、提示コードではcontainWordが２個以上あると意図通りに動かないと思います。

def is_ng(c):
    return c in 'ac'

def is_ok(c):
    return c in 'bd'

cont_cnt = 0
ret = []
for i,c in enumerate('abcdbfbdad'):
    print((i,c), end='=')

    if is_ng(c):
        print('ng')
        cont_cnt = 0
    elif is_ok(c):
        cont_cnt += 1
        if cont_cnt <= 1:
            print('output')
            ret.append((i,c))
        else:
            print('continuous')
    else:
        print('not ok')
        cont_cnt = 0

print(ret) # [(1, 'b'), (3, 'd'), (6, 'b'), (9, 'd')]

Answer 3

指定した文字列が含まれた行が検出されたときに、なにかフラグを立てるようにしといて、
指定した文字列が含まれた行が検出されたときに、そのフラグが立ってたら、その行を削除する、ってのでどうでしょう