一些方便自己批量处理文件的Python函数

一些方便自己处理文件的Python函数
用python批量处理文件(复制、重命名比较方便)
但是很多函数的用法就是记不住,因此在这里记录一些 以后免得四处去百度

import os
from natsort import natsorted # 用于文件名的自然排序
from tqdm import tqdm # 用于精度条打印
import shutil # 用于复制文件

source_dir = '/data/timer/Comparison/Reg/MSRS/elastic1/vi' # 源文件所在的文件夹
save_dir = '/data/timer/Comparison/Reg/MSRS/elastic_running_time/vi' # 目标文件夹
os.makedirs(save_dir, exist_ok=True) # 如果目标文件夹不存在则新建
landmark_dir = '/data/timer/Comparison/Reg/MSRS/landmarks' #用于遍历文件的文件夹
file_list = natsorted(os.listdir(landmark_dir)) #枚举出需要遍历的文件 其中natsorted 用于对文件命进行自然排序 
# 有的文件命可能会按1 10 这样来排序很恶心 所以推荐使用natsorted

# file_list = os.listdir(landmark_dir)
file_bar = tqdm(enumerate(file_list), total=len(file_list)) #tqdm用于打印进度条

for i, item in file_bar:
    source_name = os.path.join(source_dir, item.replace('_lms.mat', '.png')) # 获取源文件的路径
    save_name = os.path.join(save_dir, item.replace('_lms.mat', '.png')) #构造需要保存的文件路径
    file_bar.set_description(f'Moving {source_name} to {save_name}') #设置进度条的信息
    shutil.copy(source_name, save_name) #复制文件
print(save_name)
        

如有疑问可联系:[email protected]; 备注 姓名+学校

猜你喜欢

转载自blog.csdn.net/fovever_/article/details/127560594