Python 字典 setdefault()方法

【摘要】 Python 字典 setdefault()方法

Python 字典 setdefault()

Python 字典 setdefault() 函数和 get()方法 类似, 如果键不存在于字典中,将会添加键并将值设为默认值。

dict.setdefault(key, default=None)
参数

  1. key – 查找的键值。
  2. default – 键不存在时,设置的默认键值。
  3. 返回值
    如果字典中包含有给定键,则返回该键对应的值,否则返回为该键设置的值。

实例:统计当前文件夹下的各类文件的总数

import os
def my_file():
    my_dict = {}
    value = 0
    all_files = os.listdir(os.curdir)
    for each_file in all_files:
        if os.path.isdir(each_file):
            my_dict.setdefault("文件夹",0)
            my_dict["文件夹"]+=1
        else:
            path,ext = os.path.splitext(each_file)
            my_dict.setdefault(ext,0)
            my_dict[ext]+=1
    for  k in my_dict:
        print("{}:{}".format(k,my_dict[k]))
my_file()
© 版权声明
THE END
喜欢就支持一下吧
点赞0 分享