使用Python分类本地猫狗图片失败,代码问题求助
问题排查与修正代码
代码中的核心问题
- 目标路径被覆盖:你先后给
destination_dir赋值了猫和狗的文件夹路径,后面的赋值会直接覆盖前面的,导致所有文件最终都只会被移到狗的文件夹。 - 判断逻辑错误:你用完整文件路径(
os.path.join(source_dir, file_name_i))去判断是否匹配output中的序列号,但实际上应该提取文件名里的序列号(去掉后缀)来做匹配。 - shutil.move参数错误:该函数的第三个参数不是目标文件夹名称,而是可选的复制函数(默认是
shutil.copy2),你错误传入的"cats"/"dogs"会导致参数异常。
修正后的代码
import os import shutil # 源文件夹路径 source_dir = "C:\\ \\file1" # 分别定义猫和狗的目标文件夹 dest_cats = "C:\\ cats" dest_dogs = "C:\\ dogs" # 猫对应的序列号元组 cat_serials = ("123344", "345566", "8888887") # 遍历源文件夹中的所有文件 for file_name in os.listdir(source_dir): # 拼接完整文件路径 file_path = os.path.join(source_dir, file_name) # 跳过文件夹,只处理文件 if not os.path.isfile(file_path): continue # 提取文件名中的序列号(去掉后缀) serial = os.path.splitext(file_name)[0] # 判断序列号是否属于猫的集合 if serial in cat_serials: shutil.move(file_path, dest_cats) else: shutil.move(file_path, dest_dogs)
关键修改说明
- 拆分目标路径:用
dest_cats和dest_dogs分别存储两个目标文件夹的路径,避免变量覆盖。 - 正确提取序列号:用
os.path.splitext(file_name)[0]获取不带后缀的文件名(即序列号),再判断是否在猫的序列号元组中。 - 修正shutil.move调用:直接传入对应的目标文件夹路径,去掉多余的第三个参数。
- 增加文件判断:跳过源文件夹中的子文件夹,只处理图片文件。
内容的提问来源于stack exchange,提问作者Ricardo Mitchell
相关产品推荐
相关产品推荐

