Python之获取指定深度目录

import os

def get_dirs_depth(path, depth): 
    current_dirs = [root for root, _, _ in os.walk(path) if str(root).count("/")==depth]
    return current_dirs or [""] # 如果没有则返回空列表

!tree work # 显示文件目录树（jupyter中运行cmd命令加“!”）
path='work'
depth=2
get_dirs_depth(path, depth)

##结果如下：
'''
work
├── Untitled Folder
├── Untitled Folder 1
│   └── Untitled Folder
│       ├── Untitled Folder
│       └── untitled.txt
├── Untitled Folder 2
│   └── Untitled Folder
└── Untitled Folder 3
    ├── Untitled Folder
    └── Untitled Folder 1

9 directories, 1 file
['work/Untitled Folder 3/Untitled Folder 1',
 'work/Untitled Folder 3/Untitled Folder',
 'work/Untitled Folder 2/Untitled Folder',
 'work/Untitled Folder/.ipynb_checkpoints',
 'work/Untitled Folder 1/Untitled Folder']
'''