對于對目錄結構進行操作和管理的程序員來說,獲取目錄樹是必不可少的需求之一。Python的os模塊和os.path模塊提供了一系列現成的API來獲取目錄樹。
import os def get_tree(path): tree = {} for root, dirs, files in os.walk(path): level = root.replace(path, '').count(os.sep) current = tree for dir in dirs: current = current.setdefault(dir, {}) for file in files: current[file] = None return tree path = '/usr/local' print(get_tree(path))
以上代碼通過遞歸遍歷目錄樹,創建一個字典來記錄文件夾和文件的關系,最終返回一個表示目錄樹的字典結構。通過該方法,我們可以獲取到整個目錄樹結構,包括所有的文件和文件夾。
目錄樹可以用于很多場景,比如文件索引和搜索等。Python內置的目錄樹獲取方法提供了便捷的實現方式,而且通過對字典結構的處理,還可以實現很多其他的功能。