在处理分类问题制作标签的时候用到了指定文件夹下的文件名获取,在这里做个小结:
OS模块是python中一个常用于处理文件信息的模块,因此相关文件的处理都会用到OS模块。获取文件名常用到的函数为
os.walk((top, topdown = True, onerror = None, followlinks = False))
具体使用方法可 参考 写的非常详细。
os.listdir
- 获取指定目录下的文件夹名(不包含子目录)
path = '/opt'
dirs = os.listdir(path)
for dir in dirs:
print dir
- 获取目录中的所有文件夹以及文件名
import os
def file_name(file_dir):
for root, dirs, files in os.walk(file_dir):
print(root) #当前目录路径
print(dirs) #当前路径下所有子目录
print(files) #当前路径下所有非目录子文件