python批量修改文件

场景：一个文件夹下有01,02，...不同序号的文件夹，每个文件夹下有一个txt（例如0.25.txt，代表一个参数）。

目标：将01下的txt文件用01_camera.txt代替，将参数值（0.25）写入新建的txt文件夹内，最后将原txt（0.25.txt）删除。

以下python代码是3.0以上的代码：

import os;

def main():
        rootDir = 'D:\Rocking_chair';
        list_dirs = os.listdir(rootDir);
        for folder_i in list_dirs:
                #change folder name
                folder_i_path = os.path.join(rootDir, folder_i);
                files_in_folder = os.listdir(folder_i_path);
                for file in files_in_folder:
                        if os.path.splitext(file)[1]=='.txt':
                                print(file);
                                new_txt_file = open(folder_i_path+'\\Rocking_chair_'+str(folder_i)+'_camera.txt', 'w');
                                new_txt_file.write(str(os.path.splitext(file)[0]));
                                new_txt_file.close();
                                os.remove(folder_i_path+'\\'+file);

if __name__=='__main__':
        main()