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()