Python搜索文件夹并制作字典

Question

我需要从某些文件夹层次结构中创建一个字典。

我在/home/Desktop/songs 中有一个文件夹 --- 里面有文件夹:“A”和另一个“B”。文件夹A和B内有文件夹:“1”和“2”

我想获取“songs”内的文件夹并获取 A 和 B 内的文件夹。然后和他们一起制作一本字典。像这样:

A={'1','2'}
B={'1','2'}

这样，如果我更新文件夹，脚本就会更新。

我开始寻找这样的东西:

os.chdir('/home/Desktop/songs')
letter = [d for d in os.listdir('.') if os.path.isdir(d)]
print letter
-----> A,B

但我不知道如何将它们附加到字典中。

Answer 1

要获取/home/Desktop/songs 下文件夹的后代，您只需迭代使用您的理解识别的文件夹的子代即可。作为一个简单的示例，(尽管您可能希望通过使用适当的循环来使其更具可读性):

import os

os.chdir('/home/Desktop/songs')
letter = {d : [child 
                  for child in os.listdir('./' + d) 
                  if os.path.isdir('./' + d)
              ] 
              for d in os.listdir('.') 
              if os.path.isdir(d)
         }
print letter

输出:

{'A': ['1', '2'], 'B': ['1', '2']}