105 lines
3.3 KiB
Python
105 lines
3.3 KiB
Python
import os
|
||
import shutil
|
||
import math
|
||
|
||
# ================= 配置区域 =================
|
||
# 【全局变量】在这里修改你要分成的份数
|
||
N_FOLDERS = 4
|
||
|
||
# 源文件夹名称 (相对于当前脚本运行目录)
|
||
SOURCE_DIR_NAME = "camera02_save"
|
||
|
||
# 支持的图片扩展名
|
||
VALID_EXTENSIONS = ('.jpg', '.jpeg', '.png', '.bmp', '.tif', '.tiff')
|
||
|
||
|
||
# ===========================================
|
||
|
||
def split_images_to_subfolders():
|
||
# 1. 获取当前脚本所在的绝对路径
|
||
base_dir = os.getcwd()
|
||
source_path = os.path.join(base_dir, SOURCE_DIR_NAME)
|
||
|
||
# 2. 检查源文件夹是否存在
|
||
if not os.path.exists(source_path):
|
||
print(f"❌ 错误: 找不到文件夹 '{source_path}'")
|
||
print("请确保脚本在包含 'camera02_save' 的目录下运行。")
|
||
return
|
||
|
||
# 3. 获取所有图片文件列表
|
||
all_files = os.listdir(source_path)
|
||
image_files = [
|
||
f for f in all_files
|
||
if f.lower().endswith(VALID_EXTENSIONS) and os.path.isfile(os.path.join(source_path, f))
|
||
]
|
||
|
||
# 按文件名排序，保证分配顺序一致（可选）
|
||
image_files.sort()
|
||
|
||
total_count = len(image_files)
|
||
|
||
if total_count == 0:
|
||
print(f"警告: '{source_path}' 中没有找到任何图片文件。")
|
||
return
|
||
|
||
print(f"发现图片总数: {total_count}")
|
||
print(f"计划分成: {N_FOLDERS} 份")
|
||
|
||
# 4. 计算分配逻辑
|
||
# 使用轮询方式分配，确保每个文件夹数量尽可能平均
|
||
# 例如：10张图分3份 -> 4, 3, 3
|
||
|
||
created_folders = []
|
||
|
||
for index, filename in enumerate(image_files):
|
||
# 计算该图片应该去哪个子文件夹 (0 到 N_FOLDERS-1)
|
||
folder_index = index % N_FOLDERS
|
||
subfolder_name = f"part_{folder_index}"
|
||
|
||
# 构建子文件夹完整路径
|
||
dest_folder_path = os.path.join(source_path, subfolder_name)
|
||
|
||
# 如果文件夹不存在，则创建
|
||
if not os.path.exists(dest_folder_path):
|
||
os.makedirs(dest_folder_path)
|
||
created_folders.append(subfolder_name)
|
||
print(f"✅ 创建子文件夹: {subfolder_name}")
|
||
|
||
# 构建源文件和目标文件路径
|
||
src_file = os.path.join(source_path, filename)
|
||
dst_file = os.path.join(dest_folder_path, filename)
|
||
|
||
# 移动文件 (剪切)
|
||
try:
|
||
shutil.move(src_file, dst_file)
|
||
except Exception as e:
|
||
print(f"❌ 移动失败 {filename}: {e}")
|
||
|
||
# 5. 统计结果
|
||
print("\n" + "=" * 30)
|
||
print("分配完成！统计如下：")
|
||
print("=" * 30)
|
||
|
||
# 重新扫描统计每个文件夹的数量
|
||
final_counts = {}
|
||
for i in range(N_FOLDERS):
|
||
fname = f"part_{i}"
|
||
fpath = os.path.join(source_path, fname)
|
||
if os.path.exists(fpath):
|
||
count = len([f for f in os.listdir(fpath) if f.lower().endswith(VALID_EXTENSIONS)])
|
||
final_counts[fname] = count
|
||
print(f"{fname}: {count} 张图片")
|
||
|
||
print("=" * 30)
|
||
print(f"总计移动: {sum(final_counts.values())} 张图片")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# 提示用户确认
|
||
print(f"当前工作目录: {os.getcwd()}")
|
||
confirm = input(f"即将把 '{SOURCE_DIR_NAME}' 中的图片均分为 {N_FOLDERS} 份。\n是否继续？(y/n): ")
|
||
|
||
if confirm.lower() == 'y':
|
||
split_images_to_subfolders()
|
||
else:
|
||
print("操作已取消。") |