OS模块用来处理文件及目录
1、os.walk()
os.walk(top[, topdown=True[, onerror=None[, followlinks=False]]])
- top : 需要遍历的目录地址
返回的是一个三元组(root,dirs,files) :
root 所指的是当前正在遍历的这个文件夹的本身的地址
dirs 是一个 list ,内容是该文件夹中所有的目录的名字(不包括子目录)
files 同样是 list , 内容是该文件夹中所有的文件(不包括子目录)
示例:
import os
import pandas as pd
file_list = []
for root, dirs, files in os.walk(os.getcwd()):
for file in files:
if file[-3:] == 'csv':
file_list.append(file)
print(file_list)
os.getcwd()为获取当前文件夹路径
结果如下:
2、