提取所有不重复的文件后缀名

 

def suffix():  
    a = dict()
    for dirpath, dirfile, dirname in os.walk(path):
        for i in dir_name:
            #根据字典key不重复的特性，value可以相同的特性，将所有的suffix存入key中，重复的自动过滤，加快处理速度；
            #在以下提取suffix的时候，只需要提取字典的key就可以了。
            suffix_name = i.split('.')[1]
            a[suffix_name] = 1
    with open(r'suffix.txt','a') as f:
        for i in a.keys():
            f.write(i+'\n')