Python os.walk()函数控制路径遍历深度

import os

#选择路径
WorkDirectory = ['C:/Windows/System32/', 'C:/Windows/SysWow64/']
for i, o in enumerate(WorkDirectory):print('%s. %s' % (i, o))
Path = WorkDirectory[int(input(”输入路径索引“)]

#以获取一定深度的路径下的所有文件名信息为例
def Get_Path_Info(Path, Depth, Get_Files = [])
    for Abpath, Dirs, Files in os.walk(Path):
        if abs(Abpath.count(os.sep) - os.getcwd().count(os.sep)) >= Depth: Continue #如果超过深度，则忽略此次遍历
        Get_Files.extend(Files)
    return Get_Files

#以仅获取指定一级目录下的所有文件的目标为例
if __name__ == '__main__':
    ALL_Files = Get_Path_Info(Path = Path, Depth = 1)
    with open('0AAA File_Name_List.txt', 'wt', encoding='utf-8') as Info:    #写入文件名信息
        for i, o in enumerate(ALL_Files): 
            content = ('%s. %s' % (i + 1, o))
            Info.writelines(content+'\n')    #写入数字排序并引入换行
            print(content)
    
"""
abs 返回遍历深度绝对值
WorkDirectory 目录列表，以便选择
Path 路径
Depth 遍历深度
enumerate 索引化列表
"""