python如何读取文件夹下所有文件及其子文件

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

如果只想提取当前目录下的子目录的名字，则

import os
def file_name(file_dir):   
    for root, dirs, files in os.walk(file_dir):    
        print(dirs) 
        break