您的位置:

优雅地获取Python字典值的技巧

一、使用get()方法

Python的字典类型可以很快地获取到所需的数据,但如果查找的key不存在,会导致程序崩溃。Python提供了get()方法解决这个问题。


my_dict = {'name': 'Jack', 'age': 25}

# 使用get()方法获取字典值
print(my_dict.get('name'))
print(my_dict.get('country'))

# 使用get()方法,可以设置默认值,避免程序崩溃
print(my_dict.get('country', 'Unknown'))

输出结果:


Jack
None
Unknown

二、使用setdefault()方法

设置字典key和value的默认值。


my_dict = {'name': 'Jack', 'age': 25}

# 如果字典中不存在该key,就添加该key和默认value
my_dict.setdefault('country', 'Unknown')
print(my_dict)

# 如果字典中存在该key,保持原有value
my_dict.setdefault('age', 30)
print(my_dict)

输出结果:


{'name': 'Jack', 'age': 25, 'country': 'Unknown'}
{'name': 'Jack', 'age': 25, 'country': 'Unknown'}

三、使用dict.get()方法添加元素到一个字典中

获取Python字典值,同时可以动态添加新元素,意味着可以使用一个变量保存元素列表,而不是多个变量。如下代码所示:


my_list = [('name', 'Jack'), ('age', 25), ('country', 'China')]
my_dict = {}

# 使用dict.get()方法获取字典值,同时动态添加新元素到字典中
for k, v in my_list:
    my_dict[k] = my_dict.get(k, []) + [v]

print(my_dict)

输出结果:


{'name': ['Jack'], 'age': [25], 'country': ['China']}

四、使用字典解析

使用字典解析,可以很方便地一行代码搞定获取值的任务。


my_dict = {'name': 'Jack', 'age': 25, 'country': 'China'}

# 使用字典解析获取所有字典值
all_values = [value for key, value in my_dict.items()]
print(all_values)

# 只获取特定key的字典值
specific_values = [my_dict[key] for key in ('name', 'age')]
print(specific_values)

输出结果:


['Jack', 25, 'China']
['Jack', 25]

五、使用collections模块中的defaultdict()方法

使用defaultdict()方法,可以在字典中使用默认值。


from collections import defaultdict

my_dict = defaultdict(int)
my_dict['key1'] += 1

print(my_dict)

输出结果:


defaultdict(
   , {'key1': 1})

   

六、使用ChainMap对象组合多个字典

使用ChainMap对象组合多个字典,可以在多个字典中查找指定key的值。


from collections import ChainMap

dict1 = {'name': 'Jack'}
dict2 = {'age': 25}
dict3 = {'country': 'China'}
combined_dict = ChainMap(dict1, dict2, dict3)

print(combined_dict['name'])
print(combined_dict['age'])

输出结果:


Jack
25