如何获得不同目录中的所有文件名？

Question

可以说目录层次结构如下所示：

          A(root)
          |
B---------C--------D
|         |        |
fileB.h  fileC.png  fileD.py
         fileC1.jpg
          E
          |
         fileE.py

如何访问所有文档？ 或者只是获得路径。 有没有办法迭代所有？

我所做的：

path = sys.path[0]
for filename_dir in os.listdir(path):
     filename, ext = os.path.splitext(filename_dir)
     if ext == '.h':
         #do something
     elif ext == '.png'
         #do something
     .....

但是据我所知listdir只能访问程序的py文件所在的目录。

Answer 1

这仅给出目录下的目录和文件，但不递归：

import os

for filename in os.listdir(path):
    print filename

如果要列出绝对路径：

import os

def listdir_fullpath(d):
    return [os.path.join(d, f) for f in os.listdir(d)]

如果要进行递归搜索，则可以使用迭代器在每次迭代时返回3元组，包括父目录，目录列表和文件列表：

for i,j,k in os.walk('.'):
    print i, j, k

例如：

    import os

    path = sys.path[0]

    for dirname, dirnames, filenames in os.walk(path):
        for subdirname in dirnames:
            print "FOUND DIRECTORY: ", os.path.join(dirname, subdirname)
        for filename in filenames:
            print "FOUND FILE: ", os.path.join(dirname, filename)