map()

files = os.listdir(path)
files = filter(lambda x: 'png' in x, files)
# list是将map()迭代器中的数据全部取出来 转成list，这样所有os.remove 都执行了
list(map(lambda x: os.remove(path + '/' + x), files))

没有list(map(...))的话，文件夹里的png是不会被删除/执行的