删除指定目录下含指定字符串的图片

 如图，如何删除目录下含_res关键字的图片呢？

python代码如下

import os

def delete_image(directory,category):
    for root,dirs,files in os.walk(directory):
        for file in files:
            if file.endswith(".jpg") or file.endswith(".png"):
                file_path=os.path.join(root,file)
                if category in file_path:
                    os. remove(file_path)
                    print(f"{file_path} has been deleted")
            pass
        pass
    pass

dir="D:\\DL_work\\utils\\data\\test"
category="_res"
delete_image(dir,category)

 其中dir为指定根目录，category为指定关键字，执行上述代码会删除该目录下（包括子目录）所有前缀含关键词的图片（后缀为.png和.jpg），如果稍微修改，则可以删除目录下任意文件，比如

import os

def delete_image(directory,category):
    for root,dirs,files in os.walk(directory):
        for file in files:
            # if file.endswith(".jpg") or file.endswith(".png"):
            file_path=os.path.join(root,file)
            if category in file_path:
                os. remove(file_path)
                print(f"{file_path} has been deleted")
            pass
        pass
    pass

dir="D:\\DL_work\\utils\\data\\test"
category="_res"
delete_image(dir,category)

执行上面这段代码可删除目录下任意包含指定字符串的文件，比如word，不只局限于图片了