python - 如何在Python中获取目录下的文件类型

Question

我有这样的目录结构

Root\
    DirA\
        DirX\
            File.aaa
            File.bbb
        File.ccc
        File.ddd
    DirB\
        File.aaa
        File.ccc
        File.ddd
        File.eee
    File.fff

使用 python 我想获取所有文件类型的列表，如下所示:

['aaa','bbb','ccc','ddd','eee','fff']

Answer 1

m. samy's answer是正确的。然而，使用一个可以神奇地检查重复项的集合会更有效。

import os

def get_file_types(directory):
    file_ext = set()
    for root, directories, files in os.walk(directory):
        for filename in files:
            filepath = os.path.join(root, filename)
            fileName, fileExtension = os.path.splitext(filepath)


    return sorted(file_ext)  # Self-explanatory.


file_types = get_file_types("d:\\Development\\")
for ty in file_types:
    print ty