将结果保存在与输入文件同名的文件中

Question

我正在处理一个python代码，如下所示：

import os
count = 0
for doc in os.listdir('/home/krupa/Krupa/Mirellas_Image_Annotation_Data/Test/Html_Files/Texts'):
   if doc.endswith(".txt"):
      with open(doc, 'r') as f:
          single_line = ''.join([line for line in f])
          single_space = ' '.join(single_line.split())

      with open(doc.format(count) , "w") as doc:
        doc.write(single_space)
    count += 1
else:
    continue

这里我想用相同的文件名写出输出，但是扩展名不同（比如.key）。我该怎么做？请帮忙。提前致谢

Answer 1

import os
count = 0
for doc in os.listdir('/home/krupa/Krupa/Mirellas_Image_Annotation_Data/Test/Html_Files/Texts'):
    doc1 = "doc_path" + doc
    doc2 = "new_path_where_new_file_with_new_ext_needs_to_be_saved" + doc1
    if doc1.endswith(".txt"):
    with open(doc, 'r') as f:
           single_line = ''.join([line for line in f])
           single_space = ' '.join(single_line.split())

           with open(doc2.format(count) , "w") as doc2:
               doc2.write(single_space)
           count += 1
else:
continue