2012-04-03 170 views
-1

我在试图找到如何获取包含特定文件扩展名的文件夹列表,该文件夹将只给我提供文件夹并且只包含唯一文件夹。而最快的方式。Python:获取包含特定文件扩展名的文件夹

这是否有一个班轮?

所以,如果我有
C:\ folderA \ test1.tga
C:\ folderA \ test2.tga
C:\ folderA \ test3.tga
C:\ FolderB中\ test4.tga

我想
C:\ folderA \
C:\ FolderB中\

欢呼

+2

又一个“帮我解决我的功课”? – 2012-04-03 13:37:55

回答

1

有一个班轮,但它会看起来更好,如果你使用更多的线。

set(folder for folder, subfolders, files in os.walk('/') for file_ in files if os.path.splitext(file_)[1] == '.png') 
+0

谢谢,这正是我一直在寻找的! – Daniel 2012-04-03 17:01:51

1

不是一个衬垫,但:

def findFolder(root, suffix): 
    for dirName in os.listdir(root): 
    dirPath = os.path.join(root, dirName) 
    if os.path.isdir(dirPath): 
     for fileName in os.listdir(dirPath): 
     if fileName.endswith(suffix): 
      yield dirPath # or dirName, whatever you like better 
      break 
+1

当然不是单线xD – orlp 2012-04-03 13:51:27

相关问题