python实现修改文件中的内容

最近用到了modelnet数据集，由于部分.off文件格式不对，需要修改，所以写了一个小程序。

原文件：

修改之后的文件：

对的，我只把OFF单独拿出来放在了第一行，其余值没变。

代码如下：

import os

def alter(file1,file2,old_str,new_str):

     with open(file1, "r", encoding="utf-8") as f1,open(file2, "w", encoding="utf-8") as f2:
          for line in f1:
               if old_str in line:
                    line = line.replace(old_str, new_str)
               f2.write(line)



if __name__ == '__main__':
     Path = 'D:\\python_projects\\test\\table\\train'
     newpath = 'D:\\python_projects\\test\\table\\newtrain'
     dataSet &#