Python删除某些文件扩展名

我对Python相当陌生，但是我已经使这段代码可以工作，并且实际上可以执行它打算要做的事情。

但是，我想知道是否有更有效的方法对此进行编码，也许可以提高处理速度。

 import os, glob


def scandirs(path):
    for currentFile in glob.glob( os.path.join(path, '*') ):
        if os.path.isdir(currentFile):
            print 'got a directory: ' + currentFile
            scandirs(currentFile)
        print "processing file: " + currentFile
        png = "png";
        jpg = "jpg";
        if currentFile.endswith(png) or currentFile.endswith(jpg):
            os.remove(currentFile)

scandirs('C:\Program Files (x86)\music\Songs')

目前，大约有8000个文件，处理每个文件并检查它是否确实以png或jpg结尾需要花费一些时间。

最佳答案

由于您要遍历子目录，因此请使用os.walk:

import os

def scandirs(path):
    for root, dirs, files in os.walk(path):
        for currentFile in files:
            print "processing file: " + currentFile
            exts = ('.png', '.jpg')
            if currentFile.lower().endswith(exts):
                os.remove(os.path.join(root, currentFile))