文件夹均分文件图片

import os
import shutil


def split_list(lst, size):
    return [lst[i:i + size] for i in range(0, len(lst), size)]
if __name__ == '__main__':
   img_dir=r'F:\data'
   dir_name=os.path.basename(img_dir)
   img_files= ['%s/%s' % (i[0], j) for i in os.walk(img_dir) for j in i[-1] if j.endswith(('jpg', 'png', 'jpeg'))]

   groups = split_list(img_files, 5000)  # 将列表拆分成每组500个元素

   for index,group in enumerate(groups):
       target_dir=img_dir+f'/{dir_name}_{index}'
       os.makedirs(target_dir,exist_ok=True)
       print(index,target_dir)
       for img_file in group:
           json_path=img_file.replace('.jpg','.json')
           if os.path.exists(json_path):
               shutil.copy(json_path,target_dir)
           shutil.copy(img_file,target_dir)