删除指定目录下文件以及文件夹

# -*- coding: UTF-8 -*-
import os
import shutil
filelist=[]
rootdir=r"E:/aa"                      #要删除的文件夹路径
filelist=os.listdir(rootdir)
#返回指定目录下面的所有文件夹、文件名称清单（包含一级子文件夹名称，但不包括一级子文件夹内文件）
for f in filelist:
    filepath = os.path.join( rootdir, f )   #连接路径和路径下遍历的文件或一级子文件夹
    if os.path.isfile(filepath):            #判断该文件是否为文件或者文件夹
        os.remove(filepath)                 #如为文件，直接删除
        print(str(filepath)+"已删除!")
    elif os.path.isdir(filepath):
        shutil.rmtree(filepath,True)        #如为文件夹，则删除该文件夹及文件夹内所有文件
        print("dir "+str(filepath)+" 已删除!")
shutil.rmtree(rootdir,True)                 #最后删除aa总文件夹
print("删除成功！")