python递归删除指定目录

import os
import shutil
dir = "/pidancode/temp/"
def removeDir():
     for root, dirs, files in os.walk(dir, True, None, False):
         if 'xls' in root:
             print("根目录: ", root)
             files = os.listdir(root)
             for f in files:
                 if (os.path.isfile(os.path.join(root,f)) == True):
                     print('文件:', os.path.splitext(os.path.join(root,f))[0])
                     #删除文件
                     os.remove(os.path.join(root,f))
             #删除主目录        
             os.removedirs(root) 
if __name__ == '__main__':
    removeDir()