1. 程式人生 > >[python] 字典、OS庫取得當前目錄的檔案型別各幾個

[python] 字典、OS庫取得當前目錄的檔案型別各幾個

import os


all_files = os.listdir(os.curdir)
type_dict =  dict()

for each_file in all_files:
      if os.path.isdir(each_file):
            type_dict.setdefault('資料夾', 0)
            type_dict['資料夾'] += 1
      else:
            ext = os.path.splitext(each_file)[1]
            type_dict.setdefault(ext, 0)
            type_dict[ext] +=1
for each_type in type_dict.keys():
      print('該資料夾下共為型別為%s得檔案%d個' % (each_type, type_dict[each_type]))