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)

 

以上代碼實現了打印文件夾下所有文件的路徑