Python获取指定目录下所有子目录、所有文件名

需求

给出制定目录，通过Python获取指定目录下的所有子目录，所有(子目录下)文件名：

实现

import os
def file_name(file_dir):
    for root, dirs, files in os.walk(file_dir):
        print('root_dir:', root)  # 当前目录路径
        print('sub_dirs:', dirs)  # 当前路径下所有子目录
        print('files:', files)  # 当前路径下所有非目录子文件

file_name('D:\stock_data')

结果

主要用到 os 模块