程式碼如下,使用了遞迴:
1 import sys 2 currDir = sys.path[0] 3 4 import os 5 def removeFile(dir,postfix): 6 if os.path.isdir(dir): 7 for file in os.listdir(dir): 8 removeFile(dir+'/'+file,postfix) 9 else: 10 if os.path.splitext(dir)[1] == postfix: 11 os.remove(dir) 12 removeFile(currDir,'.txt')