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

Question

Lets say the directories hierarchy looks like this:

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

How can I access all the doc? Or just get the path. Is there a way to iterlate all?

What I do:

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
     .....

But as I know listdir can only access the directory where my program's py file located.

Answer 1

This gives only the dirs and files under a directory, but not recursively:

import os

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

If you want to list absolute paths:

import os

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

If you want resursive search, this gives you an iterator that returns 3-tuples including the parent directory, list of directories, and list of files at each iteration:

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

For example:

    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)