代码记录-根据一个文件夹提取另一个文件中的同名文件

#根据mask_dirs里面的文件名，将all_dirs里面的同名文件复制到sat_dirs

import os

import shutil

mask_dirs = 'D:/toWuda/BQ1228/mask/'

all_dirs = 'D:/toWuda/DOM/CGdom-114(CK17-70)0.1m/'

sat_dirs = 'D:/toWuda/BQ1228/sat/'

 

def create_dir(mask_file, all_dirs, sat_dirs):

    # img_list = os.listdir(img_dirs)

    for mask_name in os.listdir(mask_file):

         for sat_name in os.listdir(all_dirs):

            if mask_name.split('.')[0] == sat_name.split('.')[0]:

                    shutil.copy(all_dirs+sat_name, sat_dirs+sat_name)

                       

if __name__=='__main__':

    print('begin to clone pics')

    dirs = create_dir(mask_dirs, all_dirs, sat_dirs)

    print('finish copy !!!!!!!!!!')