Python：获取包含特定文件扩展名的文件夹

Question

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

这是否有一个班轮？

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

我想
C：\ folderA \
C：\ FolderB中\

欢呼

Answer 1

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

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

Answer 2

不是一个衬垫，但：

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