遍历目录

Posted 2021-02-22 director

import os

n = m = count =0
list_05 = []
def fun(argument):
 for i in os.listdir(argument):

    doucument = os.path.join(argument,i)
    if os.path.isdir(doucument):
        global  n
        n+=1  #目录总个数
        fun(doucument)
    else:
        global m
        m+=1   #文件总个数
        list_07 = os.path.splitext(i)
        if list_07[1] == ".py":  #找到所有python文件，也可以fetch模块去匹配
            list_05.append(i)
 return n,m,list_05
x,y,z=fun(‘./‘)
print(x,y,z)
或者使用os.walk会更简单