python 提供了一个很方便的内建函数,用于实现目录的遍历。下面我们根据简单实例来看下os.walk的用法
1. 这是我测试时模拟的路径结构,主目录walkdir , 子目录是testdir1 ,testdir2 ,同时testdir1 目录下还有1个它的子目录 testdir3,3个文件aaa ,bbb, ccc 分别放于testdir1-3
[root@localhost ~]# tree walkdir/
walkdir/
├── testdir1
│ ├── aaa
│ └── testdir3
│ └── ccc
└── testdir2
└── bbb
3 directories, 3 files
2. 调用os.walk()打开某个路径时,返回的会是1个递归的列表,假设为S . S[0] 代表当前路径,S[1] 代表该路径下的子目录,S[2],则当前路径或子目录下的文件。更形象的我们看下面例子:
>>> for s in os.walk('/root/walkdir'):
... print s
...
('/root/walkdir', ['testdir2', 'testdir1'], [])
('/root/walkdir/testdir2', [], ['bbb'])
('/root/walkdir/testdir1', ['testdir3'], ['aaa'])
('/root/walkdir/testdir1/testdir3', [], ['ccc'])
3. 如此,我们就可以根据我们实际需要,看是递归取目录,或是文件名,操作也很简单。
>>> for fpath, fdir, fname in os.walk('/root/walkdir'):
... print fpath
...
/root/walkdir
/root/walkdir/testdir2
/root/walkdir/testdir1
/root/walkdir/testdir1/testdir3
>>> for fpath, fdir, fname in os.walk('/root/walkdir'):
... print fdir
...
['testdir2', 'testdir1']
[]
['testdir3']
[]
>>> for fpath, fdir, fname in os.walk('/root/walkdir'):
... print fname
...
[]
['bbb']
['aaa']
['ccc']