小试牛刀-Python 批量创建文件夹

'''
实现：批量创建文件夹
'''
import os
# dir_root :指定要在哪个目录下批量创建文件夹
# dir_no : 设定要创建的文件夹数量
def create_flodes(dir_root,dir_no):
    if not dir_root or int(dir_no)<=0 : #表示输入的参数不合规
        print('输入的参数不合规')
        return None
    for i in range(1,dir_no+1) : #循环创建文件夹
        dir_name=os.path.join(dir_root,str(i)) #拼凑完整文件夹路径
        try:
            if os.path.exists(dir_name):  #判断是否存在，若存在先删除已有文件夹
                os.rmdir(dir_name) #若存在先删除已有文件夹

            os.mkdir(dir_name) #创建文件夹
            print('文件夹：{} 创建成功'.format(dir_name))
        except:
            print('文件夹：{} 创建失败'.format(dir_name))

#---开始执行
dir_root=r'E:\ #指定要在哪个目录下批量创建文件夹
dir_no = 20 # 设定要创建的文件夹数量
create_flodes(dir_root,dir_no)