python图片冲一个文件夹复制到另外一个文件夹

# 复制图像到另一个文件夹
# 文件所在文件夹
file_dir = 'F:\\Test\\TestA'
# 创建一个子文件存放文件
name = 'class'

file_list = os.listdir(file_dir)

for image in file_list:

    #如果图像名为B.png 则将B.png复制到F:\\Test\\TestA\\class
    if image == "B.png":
        if os.path.exists(os.path.join(file_dir,'class_name')):
            shutil.copy(os.path.join(file_dir,image), os.path.join(file_dir, 'class_name'))
        else:
            os.makedirs(os.path.join(file_dir,'class_name'))
            shutil.copy(os.path.join(file_dir, image), os.path.join(file_dir, 'class_name'))