Python怎么把文件夹png图片与jpg图片分开存放

使用Python将文件夹png图片与jpg图片分开存放

import os
import shutil


def split_file(images, jpg_images, png_images):

    # 区分jpg和png
    jpg = []
    png = []
    for f in os.listdir(images):
        if f.endswith('.jpg'):
            jpg.append(f)
        elif f.endswith('.png'):
            png.append(f)

    # 创建目标文件夹
    if not os.path.isdir(jpg_images):
        os.mkdir(jpg_images)
    if not os.path.isdir(png_images):
        os.mkdir(png_images)

    # 拷贝文件到目标文件夹
    for j in jpg:
        _jpg = os.path.join(images, j)
        shutil.copy(_jpg, jpg_images)
    for p in png:
        _png = os.path.join(images, p)
        shutil.copy(_png, png_images)


if __name__ == '__main__':
    base_filename = r'./Files' # 图片文件夹路径
    images = os.path.join(base_filename, 'Imgs')  # 加上存放图片文件夹的名称
    jpg_images = os.path.join(base_filename, 'jpg_images')  # 指定jpg图片存放的文件夹路径
    png_images = os.path.join(base_filename, 'png_mask') # 指定png图片存放的文件夹路径
    split_file(images, jpg_images, png_images)