Как заменить слова в пайтоне, начинающиеся на одну букву в файле

 letter = input("Введите начальную букву: ")

rf = open("input.txt", "r")

wf = open("output.txt", "w")

for line in rf:

    sep = ""

    for word in line.split():

        wf.write(sep)

        sep = " "

        wf.write("водка" if word[0] == letter[0] else word)

    wf.write("\n")

rf.close()

wf.close() 

 мир Ленин СССР

разрядка Никсон кокошник Брежнев колбаса медведи

Горбачёв перестройка хозрасчёт ускорение джинсы обналичка кооператив

матрёшка водка Пушкин балалайка 

 Введите начальную букву: к 

 мир Ленин СССР

разрядка Никсон водка Брежнев водка медведи

Горбачёв перестройка хозрасчёт ускорение джинсы обналичка водка

матрёшка водка Пушкин балалайка 

 Введите начальную букву: м 

 водка Ленин СССР

разрядка Никсон кокошник Брежнев колбаса водка

Горбачёв перестройка хозрасчёт ускорение джинсы обналичка кооператив

водка водка Пушкин балалайка 

 def replace_words(file_path, letter, replacement): 

    with open(file_path, 'r') as f: 

        text = f.read() 

     

    words = text.split() 

    replaced_words = [word if not word.startswith(letter) else replacement for word in words] 

     

    with open(file_path, 'w') as f: 

        f.write(' '.join(replaced_words))