由于当前文件夹下面的文件数量过多,希望能将文件按指定数量划分到多个子文件夹里。(共60个)
import glob
import os
from shutil import copy2
def copyfiles(src, dst):
'''
复制文件从原始路径到给定路径
如果给定路径不存在自动创建
'''
if not os.path.isdir(dst):
os.makedirs(dst)
copy2(src, dst)
def split(data, count):
'''
切分文件路径列表并返回一个生成器
count指切分后单个子文件夹里面文件的数量
'''
for i in range(1, len(data), count):
# 不一定均分 (count和给定文件列表长度不能整除时)
if i + count-1 > len(data):
start, end = (i-1, len(data))
else:
start, end = (i-1, i+count-1)
yield data[start:end]
def start_process(path, count):
# path 待切分的文件夹路径
# count 每个子文件夹中的文件个数
files = glob.glob(f'{path}/*') # 匹配的文件路径列表
splited_data = split(files, count)
for idx, folder in enumerate(splited_data):
name = f'data_{idx}'
for file in folder:
copyfiles(os.path.abspath(file), os.path.abspath(name))
path = r'C:\Users\Administrator\Desktop\split-img'
count = 10
start_process(path, count)