Python,怎么清空指定文件夹

Question

Python,怎么清空指定文件夹的所有图片，不包括子文件夹

Answer 1

import os  
path=r'C:\Users\rz\Desktop\新建文件夹'    #文件夹路径
del_file=['.jpg','.png','.pdf']      #不要的文件类型
all_file=os.listdir(path)  #遍历得到所有文件
for i in all_file:
    a,b=os.path.splitext(i)   #得到所有文件的后缀
    if b in del_file:
        os.remove(os.path.join(path,i))   #删除文件