[Py021] 批量删除

写脚本时候，每次删除中间文件，

一个一个的os.remove()实在太累了。

写了一个小函数，可以根据关键词批量删除对应文件。

def rm(path, regexs):
    '''
    通过匹配关键词，批量删除目标路径下的文件
    :param src: string, 目标路径
    :param regexs: list, 关键词列表
    '''
    path = os.path.abspath(src)
    os.chdir(path)
    for regex in regexs:
        pattern = re.compile(regex)
        for file in os.listdir(path):
            if re.search(pattern, file):
                os.remove(file)

使用

rm(path,[key1,key2,key3,...,keyN])