python根据图片名找到图片，并从源目录移动到目标目录

import os
import shutil

def move_file_to_new_directory(file_name):
    # 源目录的路径
    train_dir = "data/images/train"
    # 目标目录的路径
    new_dir = "data/error"

    # 构造文件的完整路径
    file_path = os.path.join(train_dir, file_name)

    if os.path.exists(file_path):
        # 移动文件到new目录
        shutil.move(file_path, new_dir)
        print(f"成功将文件{file_name}移动到new目录")
    else:
        print(f"找不到文件{file_name}")

# 测试 ，输入图片名，将图片移动
move_file_to_new_directory("791_strawberry.jpg")