Python 嵌套循环

Question

您好，我是 python 新手，有一个关于执行嵌套循环的最佳/pythonic 方法的问题。

我想将每个目录放入一个数组中，并使用该目录中包含的文件的嵌套数组。

我一直在研究 python 数组、字典、集合和元组，但不确定执行此操作的最佳方法

[注意我只想在一个级别上执行此操作，而不是递归地遍历所有目录]

目前我有一个函数将子目录的所有文件添加到一个数组中，但现在我也需要返回它们的父文件夹。

提前致谢

    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

我会使用字典来实现此目的，键将是目录和文件的值列表:

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

访问数据:

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