python遞歸遍歷目錄和子目錄下的所有文件,並將文件目錄存入列表

python遞歸遍歷目錄和子目錄下的所有文件,並將文件目錄存入列表


本文編寫了一個函數,實現遞歸查找目標目錄path和目標目錄下的多層子目錄下的所有文件,並將所有文件的path 存放於一個list列表中。

import os

def get_file(root_path,all_files=[]):
    '''
    遞歸函數,遍歷該文檔目錄和子目錄下的所有文件,獲取其path
    '''
    files = os.listdir(root_path)
    for file in files:
        if not os.path.isdir(root_path + '/' + file):   # not a dir
            all_files.append(root_path + '/' + file)
        else:  # is a dir
            get_file((root_path+'/'+file),all_files)
    return all_files

# example
path = './raw_data'
print(get_file(path))

運行結果(參考):

[’./raw_data/drawings/urls_drawings.txt’, ‘./raw_data/hentai/urls_hentai.txt’, ‘./raw_data/neutral/urls_neutral.txt’, ‘./raw_data/porn/urls_porn.txt’, ‘./raw_data/sexy/urls_sexy.txt’]
[Finished in 0.4s]

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章