python——获取路径下全部子目录,或非目录子文件

def file_name(file_dir):  
'''
获取路径下全部子目录、或非目录子文件
'''
    print('a') 
    for root, dirs, files in os.walk(file_dir):  
        print(root) #当前目录路径  
        print(dirs) #当前路径下全部子目录  
        print(files) #当前路径下全部非目录子文件