python - 如何获取不同目录中的所有文件名？

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)