1. 程式人生 > >樹形結構展示指定資料夾下所有檔案和資料夾

樹形結構展示指定資料夾下所有檔案和資料夾

1:匯入os包

import os

2:實現功能

<1>:使用os.walk(path)方法:

def use_walk(path):
    for (root, dirs, files) in os.walk(path):
        """
        root:當前的路徑
        dirs:當前路徑下的所有資料夾
        files:當前路徑下的所有檔案
        """
        print(root);
        print(dirs);
        print(files);

<2>:使用遞迴(推薦)

可以很詳細的顯示各個資料夾的所屬關係,類似cmd裡的 “tree” 命令。

def tree_dir(path,level=0):
    """
    :param path: 資料夾目錄
    :param level: 遞迴層級
    :return: None
    """
    print("-"*level+path);
    for dir in os.listdir(path):
        cur_path = os.path.join(path,dir);
        ## 判斷的當前獲得的是檔案還是資料夾
        if os.path.isdir(cur_path):
            tree_dir(cur_path,level+1);
        else:
            print("-"*(level+1)+os.path.basename(cur_path));