python生成和指定文件目录树一样的目录结构

import os

PNG_DIR = r'D:\PNG'
SAVE_ROOT_DIR = r'D:\jpg'

# 自顶向下对文件夹层层遍历
for curdir, dirs, files in os.walk(PNG_DIR):
    print("现在的目录：" + curdir)
    # print("该目录下的子目录：" + str(dirs))
    # print("该目录下的文件" + str(files))
# file_name = os.path.abspath(PNG_DIR)
# file_path = os.path.dirname(file_name)
# print(file_path)
	try:
        DIR_TREE = curdir.split('G\\')[1]
        SAVE_DIR = os.path.join(SAVE_ROOT_DIR, DIR_TREE)
        if not os.path.exists(SAVE_DIR):
            os.makedirs(SAVE_DIR)
    except IndexError as e:
        print("IndexError Details : " + str(e))