复制文件夹套娃中的图片、视频、xml、txt、json等文件到指定文件夹

import os
import shutil

def copy_son_img(): 
    
    path = r'C:/Users/EDY/Desktop/文件夹1/'#####需要取出文件的文件夹地址
    new_path=r'C:/Users/EDY/Desktop/视频'##文件取出后的存放地址

    image_suffix = ['bmp', 'jpg', 'jpeg', 'png', 'tif', 'tiff', 'dng','webp','jfif']
    # video_suffix = ['mp4', 'mov', 'avi', 'mkv', 'flv', 'h264', 'dav']
    # xml_suffix=['xml']
    #txt=['txt']
    #json=['json']
    for root, dirs, files in os.walk(path):
        for i in range(len(files)):
            if files[i][-3:] in image_suffix:###这里image_suffix可替换为需要取出的文件格式：video_suffix、xml_suffix、txt、json等
                file_path = root+'/'+files[i] 
                print(file_path) 
                new_file_path = new_path+ '/'+ files[i]  
                shutil.copy(file_path,new_file_path) 
if __name__ == '__main__':
    ####复制出文件夹套娃中的图片或xml
    copy_son_img()
    print('end')