python os模块之实现多层目录文件查找

本文针对多层目录下文件的查找。

一、利用 os模块的基本操作，并利用递归的思想实现了目录多层查找。

代码如下：

import os #dir_name: 处理文件的起始目录
def count_file(dir_name): #用于存目录的名字
    dir_list = [] #将当前目录下的所有文件列出
    file_list = os.listdir(dir_name) #遍历当前文件名列表
    for file_name in file_list: #如果是目录
        if os.path.isdir(file_name): dir_list.append(dir_name + "/"+ file_name) else: read_file(file_name) # 处理目录
    for dir_path in dir_list: os.chdir(dir_path) count_file(dir_path) #处理文件的函数
def read_file(file_name): with open(file_name) as f: text_list = f.readlines()
 print(file_name,":",text_list) dir_name = input("请输入文件夹路径：") count_file(dir_name)

文件处理函数并未实现什么功能，只是打印了文件名字和按行分割文件内容。

　

二、利用os模块的walk函数也可以达到遍历的效果

import os


def find_dir(dir_name):
    file_list = []
    for path_name, dir, files_name in os.walk(dir_name):
        for file in files_name:
            file_list.append(os.path.join(path_name, file))
    return file_list


file_list = find_dir(os.getcwd())
print(file_list)

 

以上代码实现了打印文件夹下所有文件的路径