python--获取基础文件

• os.walk()

        获取件路径、文件夹、文件

import os

#基础文件获取
path = r'../py攻防'

for filepath, folders, files in os.walk(path):
    #print(filepath,folders,files)   #os.walk 返回文件路径 文件夹 文件

    for file in files:
        print(os.path.join(filepath,file))    #os.path.join 拼接成规范的文件路径

 

• os.listdir()

        walk 会递归展开文件夹下的文件，listdir 只返回当前目录下的文件和文件夹名

import os

for file in os.listdir(path):
    if 'txt' in file:
        print(file)
    elif 'py' in file:
        print(file)