将文件夹中所有文件按照比例随机划分并复制到子文件夹中:
import os
import shutil
import random
# 定义源文件夹和三个子文件夹的路径
src_dir = '/path/to/source/directory'
dst_dir1 = '/path/to/destination/directory1'
dst_dir2 = '/path/to/destination/directory2'
dst_dir3 = '/path/to/destination/directory3'
# 创建三个子文件夹
os.makedirs(dst_dir1, exist_ok=True)
os.makedirs(dst_dir2, exist_ok=True)
os.makedirs(dst_dir3, exist_ok=True)
# 获取源文件夹中所有文件的路径
files = [os.path.join(src_dir, f) for f in os.listdir(src_dir) if os.path.isfile(os.path.join(src_dir, f))]
# 按照7:1:2的比例划分文件
num_files = len(files)
num_files1 = int(num_files * 0.7)
num_files2 = int(num_files * 0.1)
num_files3 = num_files - num_files1 - num_files2
# 随机打乱文件列表
random.shuffle(files)
# 将文件拷贝到三个子文件夹中
for i in range(num_files1):
shutil.copy(files[i], dst_dir1)
for i in range(num_files1, num_files1 + num_files2):
shutil.copy(files[i], dst_dir2)
for i in range(num_files1 + num_files2, num_files):
shutil.copy(files[i], dst_dir3)