- 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 #每遍历一个文件夹,值+1
- else:
- ext = os.path.splitext(each_file)[1] #分割扩展名
- type_dict.setdefault(ext, 0) #如果键值不存在,写入字典
- type_dict[ext] += 1 #根据键查找,每发现一个值对应+1
- for each_type in type_dict.keys(): #遍历字典
- print('该文件夹下共有类型为[%s]的文件%d个' % (each_type, type_dict[each_type] #输出字典的键值
复制代码
Python 字典(Dictionary) setdefault()方法
描述
Python 字典 setdefault() 函数和 get()方法 类似, 如果键不存在于字典中,将会添加键并将值设为默认值。
语法
setdefault() 方法语法:
dict.setdefault(key, default=None)
参数
key -- 查找的键值。
default -- 键不存在时,设置的默认键值。
返回值
如果字典中包含有给定键,则返回该键对应的值,否则返回为该键设置的值。
https://www.runoob.com/python/att-dictionary-setdefault.html