python删除当前目录下图片（也可以改改删别的）

不小心在当前目录生成了100000多张图片，结果电脑直接接近宕机。参考网上例子可以实现，这里没有使用os.walk，原因我没看到walk如何保留子目录下的内容，也许可以每个文件都判断是否为目录？那也太麻烦了，我们直接用后缀。

import os
path = './' #添加自己的相对路径
files= os.listdir(path) #得到文件夹下的所有文件名称
print(files[0:10])#看一看文件都有啥

for pic in files: #遍历文件夹
     
    if pic.endswith(".png"):           
        os.remove(path+'/'+pic)
    elif pic.endswith(".jpg"): 
        os.remove(path+'/'+pic)

这里针对于jpg和png，大家可以看看自用

经评论，现在加了一个path，自行更改