python判断目录是否存在，不存在则创建目录

import os
wjjname=input("请输入存放目录\n") #输入目标目录
if os.path.exists(wjjname):    #判断目标目录是否存在
    print("目录存在")
else:
    print("目录不存在")
    print("正在为您创建目录")
    os.mkdir(wjjname)          #如果不存在则创建目标目录
    print("目录创建完成")
    input("按回车键退出")