@[toc] 假设一个场景:统计一个列表中每一个数据的数目,要求:尽可能不设定初始值 如果想不出来,可以先看最基础的设定初始值的办法 首先需要导入库from collections import defaultdict 原
@[toc]
假设一个场景:统计一个列表中每一个数据的数目,要求:尽可能不设定初始值
如果想不出来,可以先看最基础的设定初始值的办法
- 首先需要导入库
from collections import defaultdict
原始方案
user_dict = {} users = ["bobby1", "bobby2", "bobby3", "bobby4", "bobby4", "bobby2", "bobby2"] for user in users: if user not in user_dict: user_dict[user] = 1 else: user_dict[user] += 1 print(user_dict)
改进
使用setdefault函数完成
user_dict = {} users = ["bobby1", "bobby2", "bobby3", "bobby4", "bobby4", "bobby2", "bobby2"] for user in users: user_dict.setdefault(user, 0) user_dict[user] += 1 print(user_dict)改进的改进
使用collection中的defaultdict方法
default_dict = defaultdict(int)users = ["bobby1", "bobby2", "bobby3", "bobby4", "bobby4", "bobby2", "bobby2"]for user in users:default_dict[user] += 1print(default_dict)
# 关于defaultdict方法的补充 > 在初始化并且设定一些特定值的时候可以用到 ```python def gen_default(): return { "name": "", "nums": 0 } default_dict = defaultdict(gen_default) print(default_dict["group1"])- 输出结果 defaultdict(<class 'int'>, {'bobby1': 1, 'bobby2': 3, 'bobby3': 1, 'bobby4': 2}) {'name': '', 'nums': 0}