先看一個需求

from collections import defaultdict

"""
需求: 統計user_list中字母出現的次數
"""

user_dict = {}
user_list = ['A','B','C','A','C']

# 第一種方式

for item in user_list:
  if item not in user_dict:
    user_dict[item] = 1
  else:
    user_dict[item] += 1

print(user_dict) # {'A': 2,'B': 1,'C': 3}

# 第二種方式
user_dict = {}
user_list = ['A','C']
for item in user_list:
  user_dict.setdefault(item,0) # 如果user_dict無item這個key,新增{item:0},如果有,不管. 而且這個方法效能比第一種方式好
  user_dict[item] += 1 # item的value值累加1

print(user_dict) # {'A': 2,'C': 3}

# 第三種方式.使用defaultdict
default_dict = defaultdict(int)
user_list = ['A','C']
for item in user_list:
  default_dict[item] += 1 # 使用這種方式,程式碼更簡單,效能也更好
print(default_dict) # defaultdict(<class 'int'>,{'A': 2,'C': 3})

defaultdict的使用

由上面的示例可知,defaultdict會根據建立例項引數的型別生成一個對應型別的預設值,

這對於dict型別的資料結構而言,可以很好避免KeyError型別的錯誤 .

此外,我們還可以自定義defaultdict呼叫物件的資料結構,以滿足我們實際需要

def person():
  '''自定義一個可呼叫物件'''
  return {
    'name': '','age': 0
  }
default_dict = defaultdict(person)
default_dict['p']
pass

是不是很贊....

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支援我們。