pythonで行頭に「リスト内にある都道府県名」がある行に追記し、新しいテキストファイル(test-new.txt)を作りたい

Question

実現したいこと

• 行頭に「リスト内にある都道府県名」がある行に追記し、新しいテキストファイル(test-new.txt)を作りたい

前提

テキストファイルを読み込み、行頭に都道府県名があった場合に、その行を「都道府県：都道府県名」と書き直し、行頭に都道府県名がない場合はそのまま元の行を書き込んで新しいファイルを作りたい

発生している問題

pythonや正規表現の初学者で、該当のソースコードでは二重に繰り返してしまうのですが、どうしても出来ず助言をいただけたら幸いです。

該当のソースコード

import re
import sys

pref_list = ['北海道','青森県','岩手県','宮城県','秋田県','山形県','福島県','茨城県','栃木県','群馬県','埼玉県','千葉県','東京都','神奈川県','新潟県','富山県','石川県','福井県','山梨県','長野県','岐阜県','静岡県','愛知県','三重県','滋賀県','京都府','大阪府','兵庫県','奈良県','和歌山県','鳥取県','島根県','岡山県','広島県','山口県','徳島県','香川県','愛媛県','高知県','福岡県','佐賀県','長崎県','熊本県','大分県','宮崎県','鹿児島県','沖縄県']

linechange = ""
f = open("test.txt", 'r',encoding="utf-8")

new_file_name = "test-new.txt"

outfile = open(new_file_name, "w")


for line in f:
    for pref in pref_list:
        #行頭にpref_listの県があった場合はlinechangeを書き込む
        if "r'^"+ pref +"'" in line:
            linechange = "都道府県："+pref
            outfile.write(linechange)

        #行頭にpref_listの県がなかった場合はそのままlineを書き込む
        else:
            #print("行頭"+pref+"ない")
            outfile.write(line)

f.close()
outfile.close()

あいうえお
かきくけこ
青森県

サシスセソ
たちつてと青森県
なにぬね青森県の

Answer 1

元のコードを生かすなら以下のような感じでできます。

for line in f:
    for pref in pref_list:
        if line.find(pref) == 0: # 先頭が都道府県名か
            line = "都道府県："+ line # 先頭に足し込む
            break # 見つかったので抜ける
    outfile.write(line)

"""
北海道,バター
りんご
新潟県,米
"""
"""
都道府県：北海道,バター
りんご
都道府県：新潟県,米
"""

Comments

[oyatsu8]

ありがとうございます。break # 見つかったので抜ける方法が思いつきませんでした。
参考にさせていただきます。

Answer 2

import re

pref_list = ['北海道','青森県','岩手県','宮城県','秋田県','山形県','福島県','茨城県','栃木県','群馬県','埼玉県','千葉県','東京都','神奈川県','新潟県','富山県','石川県','福井県','山梨県','長野県','岐阜県','静岡県','愛知県','三重県','滋賀県','京都府','大阪府','兵庫県','奈良県','和歌山県','鳥取県','島根県','岡山県','広島県','山口県','徳島県','香川県','愛媛県','高知県','福岡県','佐賀県','長崎県','熊本県','大分県','宮崎県','鹿児島県','沖縄県']
prefs = '|'.join(pref_list)

file_name = 'test.txt'
new_file_name = 'test-new.txt'
with open(file_name, 'r', encoding='utf-8') as fin, \
     open(new_file_name, 'w') as fout:
    for line in fin:
        if re.match(prefs, line):
            line = '都道府県：' + line
        fout.write(line)

Comments

[oyatsu8]

大変ありがとうございます。やりたいことができました。pythonも正規表現もまだよくわかっていないため、勉強を進めようと思います。