Question

Hello I am new to python and have a question about the best/pythonic way to do nested loops.

I want to go put each directory in an array with a nested array of the file contained in a that directory.

I have been looking at pythons arrays, dicts, sets and tupples and not sure of the best way to do this

[ Note I just want to do this for one level not recursively through all directories ]

Currently I have a function that adds all the files of sub-directories to an array, but now I need to return their parent folder too.

Thanks in advance

    def getffdirs():

    filedirs = []

    path = os.curdir

    for d in os.listdir(path):      

        if os.path.isdir(d):
            print("Entering " + d)

            curPath = os.path.join(path, d)
            for f in os.listdir(curPath):      

                if os.path.isfile(f):
                    print("file " + f)  
                    filedirs.append(f)     
    return filedirs

Answer 1

i’d use a dictionary for this purpose, the keys would be directories and the values lists of files:

def getffdirs():

  dirs = {}
  path = os.curdir
  for d in os.listdir(path):      

    if os.path.isdir(d):
        print("Entering " + d)
        dirs[d] = [] # add directory with empty list

        curPath = os.path.join(path, d)
        for f in os.listdir(curPath):      

            if os.path.isfile(f):
                print("file " + f)  
                dirs[d].append(f) # add files to appropriate directory
  return dirs

to access the data:

for dir,files in dirs.items(): # notice the call to dirs.items(), thats what was missing.
  print "directory: ",dir
  print "files:"
  for f in files:
    print f