python - os.walk()

发现了一个很好用的方法去获取当前目录路径、子目录、以及目录下所有文件名，是一个高效处理文件、目录方面的利器。

直接上例程：

import os

def dir_filename(dir_path): 

    for current_dir, sub_dirs, filenames in os.walk(dir_path): 

        print(current_dir) #当前目录路径 

        print(sub_dirs) #当前路径下所有子目录 

        print(filenames) #当前路径下所有非目录子文件

dir_path=os.getcwd()

dir_filename(dir_path)

拓展一下，获取指定文件类型的文件绝对路径：

import os

def dir_filename(dir_path):

    file_path_list=[]

    for current_dir, sub_dirs, filenames in os.walk(dir_path): 

        for file in filenames :

            file_type=os.path.splitext(file)[1]    # 获取每个文件的后缀名

            file_type=file_type.lower()        #将大写后缀名转化为小写

            if file_type in ('.jpg','.jpeg''):  # 判断文件后缀是否是所需的文件

                file_path_list.append(os.path.join(current_dir, file)) 

    print file_path_list

dir_path=os.getcwd()

dir_filename(dir_path)