- 函数作用
os.walk() 函数用于在目录树中遍历所有的文件及文件夹。
- 函数用法
输入:遍历地址path
输出:正在遍历的地址本身root、该地址下所有目录的名称dirs(list)、该地址下所有文件files(list)
使用格式:
root, dirs, files = os.walk(path)
- 代码示例
目录结构:
wx@user:/mnt/data/wx/dataset/VOC2007/ImageSets$ ls
Main
wx@user:/mnt/data/wx/dataset/VOC2007/ImageSets$ cd Main
wx@user:/mnt/data/wx/dataset/VOC2007/ImageSets/Main$ ls
test.txt train.txt trainval.txt val.txt
用法:
>>>import os
>>> root,files=os.walk('/mnt/data/wx/dataset/VOC2007/ImageSets')
>>> print('root:', root,'files:', files)
root: ('/mnt/data/wx/dataset/VOC2007/ImageSets', ['Main'], []) files: ('/mnt/data/wx/dataset/VOC2007/ImageSets/Main', [], ['trainval.txt', 'val.txt', 'train.txt', 'test.txt'])