Python面试题2-遍历文件

import os

def get_files(path,rule=".py"):
    all = []
    #fpathe 表示当前正在访问的文件夹路径f
    #dirs 表示该文件夹下的子目录名
    #fs 表示该文件夹下的文件
    for fpathe,dirs,fs in os.walk(path): #os.walk 是获取所有的目录
        for f in fs:
            filename = os.path.join(fpathe,f)
            if filename.endswith(rule): #判断是否是"xxx"结尾
                all.append(filename)
    return all

if __name__ == '__main__':
    a = get_files(r"E:\python_class\pytest")
    for i in a:
        print(i)