Python 删除目录中特定文件

代码如下，使用了递归：

import sys
currDir = sys.path[0]

import os
def removeFile(dir,postfix):
    if os.path.isdir(dir):
        for file in os.listdir(dir):
            removeFile(dir+'/'+file,postfix)
    else:
        if os.path.splitext(dir)[1] == postfix:
            os.remove(dir)
removeFile(currDir,'.txt')