Python 通过 dict 进行目录搜索和组织

Question

大家好，这是我最近第一次尝试进入Python的文件和操作系统部分。我正在尝试搜索一个目录然后找到所有子目录。如果该目录没有文件夹，则将所有文件添加到列表中。并通过听写来组织它们。

例如，一棵树可能看起来像这样的

• 起始路径
  • 目录 1
    • 子目录 1
    • 子目录 2
    • 子目录 3
      • 子子目录
        • 文件.jpg
        • 文件夹1
          • 文件1.jpg
          • 文件2.jpg
        • 文件夹2
          • 文件3.jpg
          • 文件4.jpg

即使 subsubdir 中有文件，也应该跳过它，因为它有文件夹。

现在，如果我知道要查找多少个目录，则通常可以使用 os.listdir 和 os.path.isdir 来执行此操作。但是，如果我希望它是动态的，则必须补偿任意数量的文件夹和子文件夹。我尝试过使用 os.walk，它会轻松找到所有文件。我遇到的唯一麻烦是使用包含文件的路径名创建所有字典。我需要按 dict 组织的文件夹名称，直到起始路径。

因此，最后，使用上面的示例，字典及其中的文件应如下所示：

dict['dir1']['subdir3']['subsubdir']['folder1'] = ['file1.jpg', 'file2.jpg']

dict['dir1']['subdir3']['subsubdir']['folder2'] = ['file3.jpg', 'file4.jpg']

希望对此提供任何帮助或组织信息的更好想法。谢谢。

Answer 1

也许你想要类似的东西：

def explore(starting_path):
  alld = {'': {}}

  for dirpath, dirnames, filenames in os.walk(starting_path):
    d = alld
    dirpath = dirpath[len(starting_path):]
    for subd in dirpath.split(os.sep):
      based = d
      d = d[subd]
    if dirnames:
      for dn in dirnames:
        d[dn] = {}
    else:
      based[subd] = filenames
  return alld['']

例如，给定一个 /tmp/a 这样：

$ ls -FR /tmp/a
b/  c/  d/

/tmp/a/b:
z/

/tmp/a/b/z:

/tmp/a/c:
za  zu

/tmp/a/d:

print explore('/tmp/a') 发出： {'c '：['za'，'zu']，'b'：{'z'：[]}，'d'：[]}。

如果这不正是您所追求的，也许您可​​以具体向我们展示其中的差异是什么？我怀疑如果需要的话，它们可能很容易修复。

Answer 2

我不知道你为什么要这样做。您应该能够使用 os.path.walk 进行处理，但如果您确实需要这样的结构，您可以这样做（未经测试）：

import os

def dirfunc(fdict, dirname, fnames):
    tmpdict = fdict
    keys = dirname.split(os.sep)[:-1]
    for k in keys:
        tmpdict = tmpdict.setdefault(k, {})

    for f in fnames:
        if os.path.isdir(f):
            return

    tmpdict[dirname] = fnames

mydict = {}
os.walk(directory_to_search, dirfunc, mydict)

此外，您不应该将变量命名为 dict 因为它是 Python 内置的。将名称 dict 重新绑定到 Python 的 dict 类型以外的名称是一个非常糟糕的主意。

编辑：编辑以修复“双最后一个键”错误并使用os.walk。

Answer 3

您想要构建数据的方式存在一个基本问题。如果 dir1/subdir1 包含子目录和文件，那么 dict['dir1']['subdir1'] 应该是列表还是字典？要使用 ...['subdir2'] 访问更多子目录，它需要是一个字典，但另一方面 dict['dir1']['subdir1']应该返回文件列表。

您要么必须从以某种方式结合这两个方面的自定义对象构建树，要么必须更改树结构以以不同的方式处理文件。