Python自动筛选文件类型并放置新文件夹

import os
import shutil

dir_files = os.listdir(os.getcwd())  # os.getcwd()获取当前路径
print('当前文件夹下共有', len(dir_files), '个文件(夹):', dir_files)
#新建一个文件夹
new_dir = input('\n请输入要新建的文件夹名：')
if not os.path.exists(new_dir):
    os.mkdir(new_dir)
# os.mkdir(new_dir)

print('\n图片文件开始复制')
cn = 0
#更改后缀即更改需要筛选的文件类型
for file in dir_files:
    if file.endswith('.jpg') or file.endswith('.png') or \
            file.endswith('.gif') or file.endswith('.webp'):
        if file.endswith('.jpg'):
            file_ext = '.jpg'
        elif file.endswith('.png'):
            file_ext = '.png'
        elif file.endswith('.gif'):
            file_ext = '.gif'
        else:
            file_ext = '.webp'
        cn += 1
        new_file = str(cn) + file_ext
        shutil.copyfile(file, new_dir + '/' + new_file)
        print(file, '已复制')

print('\n共复制了', cn, '个图片文件')
print(new_dir, '文件夹中的文件:', os.listdir(new_dir))

结果：将同目录下的图片文件筛选出，并自动放到新建的文件夹中。