先看一個需求
from collections import defaultdict
"""
需求: 統計user_list中字母出現的次數
"""
user_dict = {}
user_list = ['A', 'B', 'C', 'A', 'C', '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', 'B', 'C', 'A', 'C', '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, 'B': 1, 'C': 3}
# 第三種方式.使用defaultdict
default_dict = defaultdict(int)
user_list = ['A', 'B', 'C', 'A', 'C', 'C']
for item in user_list:
default_dict[item] += 1 # 使用這種方式,代碼更簡單,性能也更好
print(default_dict) # defaultdict(<class 'int' , {'A': 2, 'B': 1, 'C': 3}) 複制
defaultdict的使用
由上面的示例可知,defaultdict會根據建立執行個體參數的類型生成一個對應類型的預設值,
這對于dict類型的資料結構而言,可以很好避免KeyError類型的錯誤 .
此外,我們還可以自定義defaultdict調用對象的資料結構,以滿足我們實際需要
def person():
'''自定義一個可調用對象'''
return {
'name': '',
'age': 0
}
default_dict = defaultdict(person)
default_dict['p']
pass 複制
是不是很贊….
以上就是本文的全部内容，希望對大家的學習有所幫助。