如何用Python批量删除或移动指定图像？

# 批量删除指定名称的图像
root = r'D:\basic\aligned'
for file in os.listdir(root):
    if file.startswith('test'): # 删除root路径下 命名以'test'开头的图像
        os.remove(os.path.join(root, file))

src = r'D:\basic\aligned' # 原文件夹
dst = r'D:\compound\aligned' # 目标文件夹
for file in os.listdir(src):
    if file.startswith('train'): # 移动原文件夹中 命名以test开头的图像 到目标文件夹
        shutil.move(os.path.join(src, file),os.path.join(dst, file))