def pp(address,Suffix):
    #获取当前目录下所有的文件
        a=os.listdir(address)
        print(a)
        for i in a:
            #i 为每个文件的名字
            print(i)
            #判断文件后缀名是否为.jpg
            if i.endswith(Suffix):
                os.remove(r"C:\Users\admin\Desktop\test"+"\\" + i)
                print("删除成功")
    pp(r"C:\Users\admin\Desktop\test",".png")