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')