初學(xué)python，使用2種方法實(shí)現(xiàn)樹形目錄

方法1：

import os
path = os.getcwd()

def filenum(currPath):
    '''計(jì)算當(dāng)前目錄下文件數(shù)'''
    return sum([len(files)for root, dirs, files in os.walk(currPath)])
print('當(dāng)前路徑：%s' % path)
print('文件總數(shù)為：%s' % filenum(path))


def showTree(startPath):
    for root, dirs, files in os.walk(startPath):
        filecount = filenum(root)
        level = root.replace(startPath,' ').count(os.sep)
        deep = '|   ' * level + '|___ ' 
        print("%s%s" % (deep, os.path.split(root)[1]))
        for file in files:
            deep = '|   ' * (level+1) + '|___ ' 
            print("%s%s" %(deep, file))
showTree(path)

方法2：

import os

path = os.getcwd()
fist = path.count('/')
def all_file(currPath):
    files = (os.listdir(currPath))
    for file in files:
        newfile = currPath + '/' + file
        level = newfile.count('/')
        a = level - fist
        print('|  ' * a + '|__ ' + file)
        if os.path.isdir(newfile):
            all_file(newfile)


all_file(path)