python 统计当前目录下每个文件类型的个数

#!/usr/bin/env python3
#coding: utf-8
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]))